llava-hf
/

llava-1.5-13b-hf

Model card Files Files and versions Community

ybelkada HF staff commited on Dec 9, 2023

Commit

b512e39

•

1 Parent(s): 2cc78f9

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -75,6 +75,7 @@ model = LlavaForConditionalGeneration.from_pretrained(
     low_cpu_mem_usage=True,
 ).to(0)
 raw_image = Image.open(requests.get(image_file, stream=True).raw)
 inputs = processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)

     low_cpu_mem_usage=True,
 ).to(0)
+processor = AutoProcessor.from_pretrained(model_id)
 raw_image = Image.open(requests.get(image_file, stream=True).raw)
 inputs = processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)