oulinyu commited on
Commit
30d1e48
·
verified ·
1 Parent(s): 7d40708

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -45,7 +45,7 @@ from transformers import Qwen2_5_VLForConditionalGeneration, AutoTokenizer, Auto
45
  from qwen_vl_utils import process_vision_info
46
 
47
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
48
- "TencentBAC/TBAC-VLR1-3B-SFT", torch_dtype="auto", device_map="auto"
49
  )
50
 
51
  processor = AutoProcessor.from_pretrained("TencentBAC/TBAC-VLR1-3B")
 
45
  from qwen_vl_utils import process_vision_info
46
 
47
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
48
+ "TencentBAC/TBAC-VLR1-3B", torch_dtype="auto", device_map="auto"
49
  )
50
 
51
  processor = AutoProcessor.from_pretrained("TencentBAC/TBAC-VLR1-3B")