Update README.md (#2)
Browse files- Update README.md (e74a6636922814614b48ec8b7693e5bdd3ea08cf)
Co-authored-by: mike melamed <[email protected]>
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
|