mjbuehler commited on
Commit
b7a94e8
·
verified ·
1 Parent(s): 5cd6a92

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -56,8 +56,9 @@ def generate_response (text_input="Biology offers amazing",system_prompt='You ar
56
  temperature=temperature,
57
  num_beams=num_beams,
58
  top_k = top_k,
59
- top_p =top_p,
60
- num_return_sequences = num_return_sequences,
 
61
  do_sample =True,repetition_penalty=repetition_penalty,
62
  )
63
 
 
56
  temperature=temperature,
57
  num_beams=num_beams,
58
  top_k = top_k,
59
+ top_p = top_p,
60
+ num_return_sequences = num_return_sequences,
61
+ eos_token_id=eos_token_id,
62
  do_sample =True,repetition_penalty=repetition_penalty,
63
  )
64