Salesforce
/

instructblip-vicuna-7b

Image-Text-to-Text

image-captioning

Model card Files Files and versions

nielsr HF Staff commited on Jun 4, 2023

Commit

78f805a

·

1 Parent(s): 904cf0a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -43,7 +43,7 @@ inputs = processor(images=image, text=prompt, return_tensors="pt").to(device)
 outputs = model.generate(
         **inputs,
         do_sample=False,
-        num_beams=1,
         max_length=256,
         min_length=1,
         top_p=0.9,

 outputs = model.generate(
         **inputs,
         do_sample=False,
+        num_beams=5,
         max_length=256,
         min_length=1,
         top_p=0.9,