AIcell Anirudh25 commited on
Commit
6f36695
·
verified ·
1 Parent(s): 8fbc3f6

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 <Anirudh25@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +2 -2
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("AIcell/reproduce-1200")
117
- model = AutoModelForImageTextToText.from_pretrained("AIcell/reproduce-1200"
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