llava-hf
/

llava-1.5-7b-hf

Image-Text-to-Text

Model card Files Files and versions Community

ybelkada commited on Dec 9, 2023

Commit

0db0c8c

•

1 Parent(s): d5e0832

Update README.md

Files changed (1) hide show

README.md +2 -0

README.md CHANGED Viewed

@@ -76,6 +76,8 @@ 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)