Fix huggingface model checkpoint name to "turningpoint-ai/VisualThinker-R1-Zero" (#1)
Browse files- Fix huggingface model checkpoint name to "turningpoint-ai/VisualThinker-R1-Zero" (a6118183dd8cf001fa06478bcd98b1826c4a640c)
Co-authored-by: Anirudh Thatipelli <[email protected]>
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 |
|