Update generation_test_hf_script.py

generation_test_hf_script.py

@@ -33,10 +33,10 @@ def run_test(model_name, test_ds):
     print("update: model - ", model_name)    
     print("update: device - ", device)
 
-    model = AutoModelForCausalLM.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
     model.to(device)
 
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 
     for i, entries in enumerate(test_ds):