llava-hf
/

llava-v1.6-34b-hf

Image-Text-to-Text

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

merve HF staff commited on Mar 20

Commit

ef4abcf

•

1 Parent(s): e69afc2

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -38,9 +38,9 @@ import torch
 from PIL import Image
 import requests
-processor = LlavaNextProcessor.from_pretrained("NousResearch/Nous-Hermes-2-Yi-34B")
-model = LlavaNextForConditionalGeneration.from_pretrained("NousResearch/Nous-Hermes-2-Yi-34B", torch_dtype=torch.float16, low_cpu_mem_usage=True)
 model.to("cuda:0")
 # prepare image and text prompt, using the appropriate prompt template

 from PIL import Image
 import requests
+processor = LlavaNextProcessor.from_pretrained("llava-hf/llava-v1.6-34b-hf")
+model = LlavaNextForConditionalGeneration.from_pretrained("llava-hf/llava-v1.6-34b-hf", torch_dtype=torch.float16, low_cpu_mem_usage=True)
 model.to("cuda:0")
 # prepare image and text prompt, using the appropriate prompt template