doberst commited on
Commit
1a43201
1 Parent(s): 5bdef98

Update generation_test_hf_script.py

Browse files
Files changed (1) hide show
  1. generation_test_hf_script.py +2 -2
generation_test_hf_script.py CHANGED
@@ -31,10 +31,10 @@ def run_test(model_name, test_ds):
31
 
32
  print("update: model will be loaded on device - ", device)
33
 
34
- model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
35
  model.to(device)
36
 
37
- tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
38
 
39
  for i, entries in enumerate(test_ds):
40
 
 
31
 
32
  print("update: model will be loaded on device - ", device)
33
 
34
+ model = AutoModelForCausalLM.from_pretrained(model_name)
35
  model.to(device)
36
 
37
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
38
 
39
  for i, entries in enumerate(test_ds):
40