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 |
|