remove DPO from model id
Browse files
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-
|
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-
|
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,
|