NickyNicky commited on
Commit
bba97c3
·
verified ·
1 Parent(s): 9323403

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -126,10 +126,10 @@ inputs = tokenizer.encode(prompt,
126
 
127
  generation_config = GenerationConfig(
128
  max_new_tokens=700,
129
- temperature=0.5,
130
- top_p=0.9,
131
- top_k=40,
132
- repetition_penalty=1.1, #1.1, # 1.0 means no penalty, > 1.0 means penalty, 1.2 from CTRL paper
133
  do_sample=True,
134
  pad_token_id=tokenizer.eos_token_id,
135
  eos_token_id=tokenizer.eos_token_id,
 
126
 
127
  generation_config = GenerationConfig(
128
  max_new_tokens=700,
129
+ # temperature=0.5,
130
+ # top_p=0.9,
131
+ # top_k=40,
132
+ # repetition_penalty=1.1, #1.1, # 1.0 means no penalty, > 1.0 means penalty, 1.2 from CTRL paper
133
  do_sample=True,
134
  pad_token_id=tokenizer.eos_token_id,
135
  eos_token_id=tokenizer.eos_token_id,