Update generation_test_hf_script.py
Browse files
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
|
35 |
model.to(device)
|
36 |
|
37 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name
|
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 |
|