add max_new_tokens (#8)
Browse files- add max_new_tokens (68d8e1aee5967f9d192de45970bd7d1a7a79102f)
- generation_config.json +1 -0
generation_config.json
CHANGED
@@ -3,5 +3,6 @@
|
|
3 |
"bos_token_id": 1,
|
4 |
"eos_token_id": 2,
|
5 |
"pad_token_id": 2,
|
|
|
6 |
"transformers_version": "4.42.3"
|
7 |
}
|
|
|
3 |
"bos_token_id": 1,
|
4 |
"eos_token_id": 2,
|
5 |
"pad_token_id": 2,
|
6 |
+
"max_new_tokens": 40,
|
7 |
"transformers_version": "4.42.3"
|
8 |
}
|