Update README.md
Browse files
README.md
CHANGED
|
@@ -29,7 +29,7 @@ summary_text_ids = model.generate(
|
|
| 29 |
input_ids=input_ids,
|
| 30 |
bos_token_id=model.config.bos_token_id,
|
| 31 |
eos_token_id=model.config.eos_token_id,
|
| 32 |
-
length_penalty=
|
| 33 |
max_length=256,
|
| 34 |
min_length=12,
|
| 35 |
num_beams=6,
|
|
|
|
| 29 |
input_ids=input_ids,
|
| 30 |
bos_token_id=model.config.bos_token_id,
|
| 31 |
eos_token_id=model.config.eos_token_id,
|
| 32 |
+
length_penalty=1.5,
|
| 33 |
max_length=256,
|
| 34 |
min_length=12,
|
| 35 |
num_beams=6,
|