prithivMLmods commited on
Commit
f41bac5
·
verified ·
1 Parent(s): c817b96

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -63,21 +63,21 @@ from qwen_vl_utils import process_vision_info
63
 
64
  # Load the model
65
  model = Qwen2VLForConditionalGeneration.from_pretrained(
66
- "prithivMLmods/Callisto-OCR3-2B-Instruct", # replace with updated model ID if available
67
  torch_dtype="auto",
68
  device_map="auto"
69
  )
70
 
71
  # Optional: Flash Attention for performance optimization
72
  # model = Qwen2VLForConditionalGeneration.from_pretrained(
73
- # "prithivMLmods/Callisto-OCR3-2B-Instruct",
74
  # torch_dtype=torch.bfloat16,
75
  # attn_implementation="flash_attention_2",
76
  # device_map="auto",
77
  # )
78
 
79
  # Load processor
80
- processor = AutoProcessor.from_pretrained("prithivMLmods/Callisto-OCR3-2B-Instruct")
81
 
82
  messages = [
83
  {
 
63
 
64
  # Load the model
65
  model = Qwen2VLForConditionalGeneration.from_pretrained(
66
+ "prithivMLmods/Imgscope-OCR-2B-0527", # replace with updated model ID if available
67
  torch_dtype="auto",
68
  device_map="auto"
69
  )
70
 
71
  # Optional: Flash Attention for performance optimization
72
  # model = Qwen2VLForConditionalGeneration.from_pretrained(
73
+ # "prithivMLmods/Imgscope-OCR-2B-0527",
74
  # torch_dtype=torch.bfloat16,
75
  # attn_implementation="flash_attention_2",
76
  # device_map="auto",
77
  # )
78
 
79
  # Load processor
80
+ processor = AutoProcessor.from_pretrained("prithivMLmods/Imgscope-OCR-2B-0527")
81
 
82
  messages = [
83
  {