aadya1762 commited on
Commit
51cf9c8
·
1 Parent(s): 9038c58
Files changed (1) hide show
  1. 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(