baconnier commited on
Commit
4dffd30
·
verified ·
1 Parent(s): 5826b06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -84,6 +84,7 @@ model = AutoModelForCausalLM.from_pretrained(
84
  config=config,
85
  device_map="auto",
86
  torch_dtype=torch.bfloat16,
 
87
  )
88
  model.eval()
89
 
 
84
  config=config,
85
  device_map="auto",
86
  torch_dtype=torch.bfloat16,
87
+ trust_remote_code=True
88
  )
89
  model.eval()
90