amazingvince
/

custom-seq2seq

Model card Files Files and versions

amazingvince commited on Sep 3, 2024

Commit

185177f

·

verified ·

1 Parent(s): b895df3

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -4,8 +4,8 @@
     "custom_seq2seq_llm"
   ],
   "auto_map": {
-    "AutoConfig": "configuration_custom_seq2seq_llm.py",
-    "AutoModelForSeq2SeqLM": "modeling_custom_seq2seq_llm.py"
   },
   "attention_probs_dropout_prob": 0.0,
   "bos_token_id": 1,

     "custom_seq2seq_llm"
   ],
   "auto_map": {
+    "AutoConfig": "configuration_custom_seq2seq_llm.Seq2SeqConfig",
+    "AutoModelForSeq2SeqLM": "modeling_custom_seq2seq_llm.CustomSeq2SeqLLM"
   },
   "attention_probs_dropout_prob": 0.0,
   "bos_token_id": 1,