imryanxu commited on
Commit
84b01b5
·
verified ·
1 Parent(s): 3901cc3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -11,7 +11,7 @@ from qwen_vl_utils import process_vision_info
11
  import torch
12
 
13
  # We recommend enabling flash_attention_2 for better acceleration and memory saving.
14
- model_dir = "Qwen2.5-VL-7B-Instruct"
15
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
16
  model_dir,
17
  torch_dtype=torch.bfloat16,
 
11
  import torch
12
 
13
  # We recommend enabling flash_attention_2 for better acceleration and memory saving.
14
+ model_dir = "Qwen2.5-VL-7B-Instruct-sft"
15
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
16
  model_dir,
17
  torch_dtype=torch.bfloat16,