Update README.md
Browse files
README.md
CHANGED
|
@@ -7,10 +7,10 @@ from qwen_vl_utils import process_vision_info
|
|
| 7 |
instruct_prompt = r"You FIRST think about the reasoning process as an internal monologue and then provide the final answer. The reasoning process MUST BE enclosed within <think> </think> tags. The final answer MUST BE put in \boxed{}."
|
| 8 |
|
| 9 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
| 10 |
-
"
|
| 11 |
)
|
| 12 |
|
| 13 |
-
processor = AutoProcessor.from_pretrained("
|
| 14 |
|
| 15 |
greedy_generation_config = GenerationConfig(
|
| 16 |
do_sample=False,
|
|
|
|
| 7 |
instruct_prompt = r"You FIRST think about the reasoning process as an internal monologue and then provide the final answer. The reasoning process MUST BE enclosed within <think> </think> tags. The final answer MUST BE put in \boxed{}."
|
| 8 |
|
| 9 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
| 10 |
+
"furonghuang-lab/ThinkLite-VL-7B", torch_dtype="auto", device_map="auto"
|
| 11 |
)
|
| 12 |
|
| 13 |
+
processor = AutoProcessor.from_pretrained("furonghuang-lab/ThinkLite-VL-7B")
|
| 14 |
|
| 15 |
greedy_generation_config = GenerationConfig(
|
| 16 |
do_sample=False,
|