Yehor commited on
Commit
12015a7
·
verified ·
1 Parent(s): 049a322

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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