RaushanTurganbay HF staff commited on
Commit
cc44dd6
1 Parent(s): 799fe84

remove DPO from model id

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -44,7 +44,7 @@ from transformers import pipeline
44
  from PIL import Image
45
  import requests
46
 
47
- model_id = "llava-hf/llava-interleave-qwen-0.5b-dpo-hf"
48
  pipe = pipeline("image-to-text", model=model_id)
49
 
50
  url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/ai2d-demo.jpg"
@@ -79,7 +79,7 @@ from PIL import Image
79
  import torch
80
  from transformers import AutoProcessor, LlavaForConditionalGeneration
81
 
82
- model_id = "llava-hf/llava-interleave-qwen-0.5b-dpo-hf"
83
  model = LlavaForConditionalGeneration.from_pretrained(
84
  model_id,
85
  torch_dtype=torch.float16,
 
44
  from PIL import Image
45
  import requests
46
 
47
+ model_id = "llava-hf/llava-interleave-qwen-0.5b-hf"
48
  pipe = pipeline("image-to-text", model=model_id)
49
 
50
  url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/ai2d-demo.jpg"
 
79
  import torch
80
  from transformers import AutoProcessor, LlavaForConditionalGeneration
81
 
82
+ model_id = "llava-hf/llava-interleave-qwen-0.5b-hf"
83
  model = LlavaForConditionalGeneration.from_pretrained(
84
  model_id,
85
  torch_dtype=torch.float16,