Spaces:
Paused
Paused
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 |
|