usmankhanic
/

apex-seq2seq

encoder-decoder

Model card Files Files and versions

usmankhanic commited on Feb 16, 2025

Commit

3e46063

·

verified ·

1 Parent(s): b19bc48

Publishing model weights bin

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -4,7 +4,7 @@
   ],
   "model_type": "encoder-decoder",
   "encoder": {
-    "model_type": "custom-encoder",
     "num_layers": 3,
     "hidden_size": 256,
     "num_attention_heads": 8,
@@ -19,7 +19,7 @@
     }
   },
   "decoder": {
-    "model_type": "custom-decoder",
     "num_layers": 3,
     "hidden_size": 256,
     "num_attention_heads": 8,

   ],
   "model_type": "encoder-decoder",
   "encoder": {
+    "model_type": "bart",
     "num_layers": 3,
     "hidden_size": 256,
     "num_attention_heads": 8,
     }
   },
   "decoder": {
+    "model_type": "bart",
     "num_layers": 3,
     "hidden_size": 256,
     "num_attention_heads": 8,