Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -15,6 +15,11 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
15 |
torch_dtype='auto',
|
16 |
)
|
17 |
|
|
|
|
|
|
|
|
|
|
|
18 |
print('Model dtype:', model.dtype)
|
19 |
|
20 |
|
|
|
15 |
torch_dtype='auto',
|
16 |
)
|
17 |
|
18 |
+
model = torch.compile(model, backend="eager")
|
19 |
+
|
20 |
+
# import torch._dynamo
|
21 |
+
# torch._dynamo.config.suppress_errors = True
|
22 |
+
|
23 |
print('Model dtype:', model.dtype)
|
24 |
|
25 |
|