nielsr HF staff unk1911 commited on
Commit
f898b99
1 Parent(s): c892f3e

Update README.md (#2)

Browse files

- Update README.md (e74a6636922814614b48ec8b7693e5bdd3ea08cf)


Co-authored-by: mike melamed <unk1911@users.noreply.huggingface.co>

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