Spaces:
Build error
Build error
Update app.py
Browse files
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 |
|