Commit
•
58040e9
1
Parent(s):
f5347fd
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -63,7 +63,7 @@ def inference_and_run(image_path, prompt, conv_mode="ferret_llama_3", model_path
|
|
| 63 |
"--image_path", ".",
|
| 64 |
"--answers_file", "eval_output.jsonl",
|
| 65 |
"--num_beam", "1",
|
| 66 |
-
"--max_new_tokens", "
|
| 67 |
"--conv_mode", conv_mode
|
| 68 |
]
|
| 69 |
|
|
|
|
| 63 |
"--image_path", ".",
|
| 64 |
"--answers_file", "eval_output.jsonl",
|
| 65 |
"--num_beam", "1",
|
| 66 |
+
"--max_new_tokens", "1024",
|
| 67 |
"--conv_mode", conv_mode
|
| 68 |
]
|
| 69 |
|