nielsr HF staff commited on
Commit
e1790f3
1 Parent(s): 8e04490

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -43,7 +43,7 @@ inputs = processor(images=image, text=prompt, return_tensors="pt").to(device)
43
  outputs = model.generate(
44
  **inputs,
45
  do_sample=False,
46
- num_beams=1,
47
  max_length=256,
48
  min_length=1,
49
  top_p=0.9,
 
43
  outputs = model.generate(
44
  **inputs,
45
  do_sample=False,
46
+ num_beams=5,
47
  max_length=256,
48
  min_length=1,
49
  top_p=0.9,