Spaces:
Sleeping
Sleeping
fix
Browse files- gemmademo/_model.py +3 -0
gemmademo/_model.py
CHANGED
|
@@ -9,6 +9,9 @@ def load_model(name: str, device_map: str = "cpu"):
|
|
| 9 |
"""
|
| 10 |
Model loading function that loads the model without caching
|
| 11 |
"""
|
|
|
|
|
|
|
|
|
|
| 12 |
tokenizer = AutoTokenizer.from_pretrained(name)
|
| 13 |
|
| 14 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
|
| 9 |
"""
|
| 10 |
Model loading function that loads the model without caching
|
| 11 |
"""
|
| 12 |
+
import torch._dynamo
|
| 13 |
+
torch._dynamo.config.suppress_errors = True
|
| 14 |
+
|
| 15 |
tokenizer = AutoTokenizer.from_pretrained(name)
|
| 16 |
|
| 17 |
model = AutoModelForCausalLM.from_pretrained(
|