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,
|