Update README.md
Browse files
README.md
CHANGED
@@ -46,7 +46,7 @@ model_inputs = tokenizer(input_text.format(prompt=prompt), return_tensors="pt").
|
|
46 |
gen_conf = {
|
47 |
"temperature": 0.7,
|
48 |
"repetition_penalty": 1.0,
|
49 |
-
"max_new_tokens":
|
50 |
"do_sample": True,
|
51 |
}
|
52 |
|
|
|
46 |
gen_conf = {
|
47 |
"temperature": 0.7,
|
48 |
"repetition_penalty": 1.0,
|
49 |
+
"max_new_tokens": 512,
|
50 |
"do_sample": True,
|
51 |
}
|
52 |
|