ybelkada commited on
Commit
b512e39
1 Parent(s): 2cc78f9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -75,6 +75,7 @@ model = LlavaForConditionalGeneration.from_pretrained(
75
  low_cpu_mem_usage=True,
76
  ).to(0)
77
 
 
78
 
79
  raw_image = Image.open(requests.get(image_file, stream=True).raw)
80
  inputs = processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)
 
75
  low_cpu_mem_usage=True,
76
  ).to(0)
77
 
78
+ processor = AutoProcessor.from_pretrained(model_id)
79
 
80
  raw_image = Image.open(requests.get(image_file, stream=True).raw)
81
  inputs = processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)