llava-hf
/

bakLlava-v1-hf

Image-Text-to-Text

Model card Files Files and versions Community

ybelkada commited on Dec 9, 2023

Commit

0f9a538

•

1 Parent(s): d169786

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

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