PerRing commited on
Commit
85a8af7
1 Parent(s): 1488efa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -27,7 +27,7 @@ image_file = "https://images.pexels.com/photos/757889/pexels-photo-757889.jpeg?a
27
  raw_image = Image.open(requests.get(image_file, stream=True).raw)
28
  inputs = processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)
29
 
30
- output = model.generate(**inputs, max_length=256)
31
  print(processor.decode(output[0], skip_special_tokens=True))
32
  ```
33
  ## result
 
27
  raw_image = Image.open(requests.get(image_file, stream=True).raw)
28
  inputs = processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)
29
 
30
+ output = model.generate(**inputs, max_length=256, temperature=0.4, do_sample=True)
31
  print(processor.decode(output[0], skip_special_tokens=True))
32
  ```
33
  ## result