gokaygokay commited on
Commit
8c73579
·
verified ·
1 Parent(s): 37228ac

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -28,7 +28,7 @@ model_inputs = processor(text=prompt, images=image, return_tensors="pt")
28
  input_len = model_inputs["input_ids"].shape[-1]
29
 
30
  with torch.inference_mode():
31
- generation = model.generate(**model_inputs, max_new_tokens=100, do_sample=False)
32
  generation = generation[0][input_len:]
33
  decoded = processor.decode(generation, skip_special_tokens=True)
34
  print(decoded)
 
28
  input_len = model_inputs["input_ids"].shape[-1]
29
 
30
  with torch.inference_mode():
31
+ generation = model.generate(**model_inputs, max_new_tokens=256, do_sample=False)
32
  generation = generation[0][input_len:]
33
  decoded = processor.decode(generation, skip_special_tokens=True)
34
  print(decoded)