Update README.md
Browse filescorrected to point to the right model namespace: llava-hf/llava-v1.6-vicuna-13b-hf
README.md
CHANGED
@@ -36,9 +36,9 @@ import torch
|
|
36 |
from PIL import Image
|
37 |
import requests
|
38 |
|
39 |
-
processor = LlavaNextProcessor.from_pretrained("llava-v1.6-vicuna-13b-hf")
|
40 |
|
41 |
-
model = LlavaNextForConditionalGeneration.from_pretrained("llava-v1.6-vicuna-13b-hf", torch_dtype=torch.float16, low_cpu_mem_usage=True)
|
42 |
model.to("cuda:0")
|
43 |
|
44 |
# prepare image and text prompt, using the appropriate prompt template
|
|
|
36 |
from PIL import Image
|
37 |
import requests
|
38 |
|
39 |
+
processor = LlavaNextProcessor.from_pretrained("llava-hf/llava-v1.6-vicuna-13b-hf")
|
40 |
|
41 |
+
model = LlavaNextForConditionalGeneration.from_pretrained("llava-hf/llava-v1.6-vicuna-13b-hf", torch_dtype=torch.float16, low_cpu_mem_usage=True)
|
42 |
model.to("cuda:0")
|
43 |
|
44 |
# prepare image and text prompt, using the appropriate prompt template
|