NickyNicky
commited on
Update README.md
Browse files
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,
|