Thaweewat commited on
Commit
2799635
1 Parent(s): c9c789a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -60,7 +60,7 @@ def generate_response(input_text: str) -> str:
60
  generation_config = GenerationConfig(
61
  do_sample=True,
62
  top_k=1,
63
- temperature=0.5,
64
  max_new_tokens=300,
65
  repetition_penalty=1.1,
66
  pad_token_id=tokenizer.eos_token_id)
 
60
  generation_config = GenerationConfig(
61
  do_sample=True,
62
  top_k=1,
63
+ temperature=0.4, # After a few experiment I found that between 0.3-0.4 seem to generate well
64
  max_new_tokens=300,
65
  repetition_penalty=1.1,
66
  pad_token_id=tokenizer.eos_token_id)