Fix..made it finally work

config.json

@@ -18,7 +18,8 @@
   "hidden_size": 16,
   "init_std": 0.02,
   "is_encoder_decoder": true,
-  "max_position_embeddings": 512,
+  "max_length": 64,
+  "max_position_embeddings": 64,
   "model_type": "prophetnet",
   "ngram": 2,
   "num_buckets": 32,
@@ -31,5 +32,5 @@
   "torch_dtype": "float32",
   "transformers_version": "4.27.1",
   "use_cache": true,
-  "vocab_size": 512
+  "vocab_size": 30522
 }

generation_config.json

@@ -3,6 +3,7 @@
   "bos_token_id": 1,
   "decoder_start_token_id": 0,
   "eos_token_id": 2,
+  "max_length": 64,
   "pad_token_id": 0,
   "transformers_version": "4.27.1"
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:480db482a3109c08731ea97961ddc33f61f53990e27f524d85f8a28fc4640e3d
-size 284013
+oid sha256:83891082b506f21e650629d0ec1e05f31eeb2911106c743c4b1c71c97597df18
+size 2147309