NicoZenith
/

onevision-7b-all-vqa-conv

llava_onevision

Model card Files Files and versions Community

NicoZenith commited on about 1 month ago

Commit

010cd6a

•

1 Parent(s): 813cb73

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -35,4 +35,6 @@ raw_image = Image.open(requests.get(image_file, stream=True).raw)
 inputs = processor(images=raw_image, text=prompt, return_tensors='pt').to(0, torch.float16)
 output = model.generate(**inputs, max_new_tokens=200, do_sample=False)
-print(processor.decode(output[0][2:], skip_special_tokens=True))

 inputs = processor(images=raw_image, text=prompt, return_tensors='pt').to(0, torch.float16)
 output = model.generate(**inputs, max_new_tokens=200, do_sample=False)
+response_text = processor.decode(output[0][2:], skip_special_tokens=True)
+response_text = response_text.split("assistant\n")[-1]
+print(response_text)