Fix huggingface model checkpoint name to "turningpoint-ai/VisualThinker-R1-Zero"
#1
by
Anirudh25
- opened
README.md
CHANGED
@@ -113,8 +113,8 @@ from io import BytesIO
|
|
113 |
from transformers import AutoProcessor, AutoModelForImageTextToText
|
114 |
|
115 |
# Load model directly
|
116 |
-
processor = AutoProcessor.from_pretrained("
|
117 |
-
model = AutoModelForImageTextToText.from_pretrained("
|
118 |
,torch_dtype="auto", device_map="auto")
|
119 |
model.eval()
|
120 |
|
|
|
113 |
from transformers import AutoProcessor, AutoModelForImageTextToText
|
114 |
|
115 |
# Load model directly
|
116 |
+
processor = AutoProcessor.from_pretrained("turningpoint-ai/VisualThinker-R1-Zero")
|
117 |
+
model = AutoModelForImageTextToText.from_pretrained("turningpoint-ai/VisualThinker-R1-Zero"
|
118 |
,torch_dtype="auto", device_map="auto")
|
119 |
model.eval()
|
120 |
|