Update eval.py
Browse files
eval.py
CHANGED
@@ -6,7 +6,7 @@ from transformers.generation import GenerationConfig
|
|
6 |
tokenizer = AutoTokenizer.from_pretrained("qwen/Qwen-VL-Chat",trust_remote_code=True)
|
7 |
|
8 |
model = AutoModelForCausalLM.from_pretrained(
|
9 |
-
"
|
10 |
device_map="cuda",
|
11 |
trust_remote_code=True
|
12 |
).eval()
|
|
|
6 |
tokenizer = AutoTokenizer.from_pretrained("qwen/Qwen-VL-Chat",trust_remote_code=True)
|
7 |
|
8 |
model = AutoModelForCausalLM.from_pretrained(
|
9 |
+
"AgoraX/Qwen-VL-FNCall", # path to the output directory
|
10 |
device_map="cuda",
|
11 |
trust_remote_code=True
|
12 |
).eval()
|