mohammedalmatari commited on
Commit
ac2a11f
·
verified ·
1 Parent(s): 7b2c434

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -48,13 +48,13 @@ from transformers import Qwen2_5_VLForConditionalGeneration, AutoTokenizer, Auto
48
  from qwen_vl_utils import process_vision_info
49
  # default: Load the model on the available device(s)
50
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
51
- "neuralleap/Qwen2.5-VL-7B-Instruct-Finetuned-Os-Atlas",
52
  torch_dtype=torch.bfloat16,
53
  attn_implementation="flash_attention_2",
54
  device_map="auto",
55
  )
56
  # default processer
57
- processor = AutoProcessor.from_pretrained("neuralleap/Qwen2.5-VL-7B-Instruct-Finetuned-Os-Atlas")
58
  # Example input
59
  messages = [
60
  {
 
48
  from qwen_vl_utils import process_vision_info
49
  # default: Load the model on the available device(s)
50
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
51
+ "Fintor/Qwen2.5-VL-7B-Instruct-Finetuned-Os-Atlas",
52
  torch_dtype=torch.bfloat16,
53
  attn_implementation="flash_attention_2",
54
  device_map="auto",
55
  )
56
  # default processer
57
+ processor = AutoProcessor.from_pretrained("Fintor/Qwen2.5-VL-7B-Instruct-Finetuned-Os-Atlas")
58
  # Example input
59
  messages = [
60
  {