Make sure the test beam size is stored

config.json

@@ -51,6 +51,11 @@
   "no_repeat_ngram_size": 3,
   "num_hidden_layers": 6,
   "pad_token_id": 1,
+  "task_specific_params": {
+    "summarization": {
+      "num_beams": 5
+    }
+  },
   "torch_dtype": "float32",
   "transformers_version": "4.19.2",
   "use_cache": true,