Hugging Face's logo

llava-hf
/
llava-v1.6-vicuna-13b-hf

Image-Text-to-Text
Transformers
Safetensors
English
llava_next
vision
text-generation-inference
Inference Endpoints
Model card Files Community
6

Update README.md

#2
by unk1911 - opened
base: refs/heads/main
←
from: refs/pr/2
Discussion Files changed
+2
-2
Files changed (1) hide show
  1. README.md +2 -2
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