Update README.md

README.md

@@ -59,7 +59,7 @@ for k,v in inputs.items():
 MAX_LENGTH = 200
 # Autoregressively generate
 # We use greedy decoding here, for more fancy methods see https://huggingface.co/blog/how-to-generate
-generated_ids = base_model.generate(**inputs, max_new_tokens=MAX_LENGTH)
+generated_ids = base_model.generate(**inputs, max_new_tokens=MAX_LENGTH,temperature=0.7,repetition_penalty=2.0,do_sample=True)
 
 # Next we turn each predicted token ID back into a string using the decode method
 # We chop of the prompt, which consists of image tokens and our text prompt
@@ -98,7 +98,7 @@ for k,v in inputs.items():
 MAX_LENGTH = 200
 # Autoregressively generate
 # We use greedy decoding here, for more fancy methods see https://huggingface.co/blog/how-to-generate
-generated_ids = base_model.generate(**inputs, max_new_tokens=MAX_LENGTH)
+generated_ids = base_model.generate(**inputs, max_new_tokens=MAX_LENGTH,temperature=0.7,repetition_penalty=2.0,do_sample=True)
 
 # Next we turn each predicted token ID back into a string using the decode method
 # We chop of the prompt, which consists of image tokens and our text prompt