Adding generation config file(s)

2023-01-24 15:02:18 +00:00 · 2023-01-24 15:02:18 +00:00 · 42ab86cf83
parent 071bd9495c
commit 42ab86cf83
1 changed files with 16 additions and 0 deletions
--- a/generation_config.json
+++ b/generation_config.json
@ -0,0 +1,16 @@
+{
+  "_from_model_config": true,
+  "bad_words_ids": [
+    [
+      65000
+    ]
+  ],
+  "bos_token_id": 0,
+  "decoder_start_token_id": 65000,
+  "eos_token_id": 0,
+  "forced_eos_token_id": 0,
+  "max_length": 512,
+  "num_beams": 4,
+  "pad_token_id": 65000,
+  "transformers_version": "4.27.0.dev0"
+}