Deeokay commited on
Commit
325d01c
·
verified ·
1 Parent(s): f5be8e3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -183,8 +183,8 @@ def generate_response2(input_text, max_length=250):
183
  max_new_tokens=max_length,
184
  attention_mask=attention_mask,
185
  do_sample=True,
186
- temperature=0.6,
187
- top_k=0,
188
  no_repeat_ngram_size=2,
189
  pad_token_id=tokenizer.pad_token_id,
190
  eos_token_id=tokenizer.eos_token_id,
 
183
  max_new_tokens=max_length,
184
  attention_mask=attention_mask,
185
  do_sample=True,
186
+ temperature=0.4, # this can be played around
187
+ top_k=60, # this can be played around
188
  no_repeat_ngram_size=2,
189
  pad_token_id=tokenizer.pad_token_id,
190
  eos_token_id=tokenizer.eos_token_id,