Fix huggingface model checkpoint name to "turningpoint-ai/VisualThinker-R1-Zero"

#1
by Anirudh25 - opened
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