Adding generation config file(s)

2023-01-24 15:10:39 +00:00 · 2023-01-24 15:10:39 +00:00 · 8e7e4bf6e9
parent afc367bc73
commit 8e7e4bf6e9
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"
+}