Upload MBartForCurveRegression

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "mittagessen/transformer_seg_decoder",
+  "_name_or_path": "mittagessen/reg_transformer_seg_decoder",
   "activation_dropout": 0.0,
   "activation_function": "gelu",
   "add_cross_attention": true,
@@ -8,7 +8,7 @@
     "MBartForCurveRegression"
   ],
   "attention_dropout": 0.0,
-  "bos_token_id": 0,
+  "bos_token_id": 1,
   "classifier_dropout": 0.0,
   "d_model": 1024,
   "decoder_attention_heads": 16,
@@ -28,7 +28,7 @@
   "max_position_embeddings": 768,
   "model_type": "mbart",
   "num_hidden_layers": 12,
-  "pad_token_id": 1,
+  "pad_token_id": 0,
   "scale_embedding": true,
   "tie_word_embeddings": false,
   "torch_dtype": "float32",