ai4bharat
/

indictrans2-indic-indic-dist-320M

text2text-generation

Model card Files Files and versions Community

jaygala24 commited on Nov 28, 2023

Commit

fa882b4

•

1 Parent(s): 6ded248

Update config.json

Files changed (1) hide show

config.json +5 -1

config.json CHANGED Viewed

@@ -1,10 +1,14 @@
 {
-  "_name_or_path": "/data/jaygala/it2_ckpts/hf_distilled_models/indic-indic",
   "activation_dropout": 0.0,
   "activation_function": "gelu",
   "architectures": [
     "IndicTransForConditionalGeneration"
   ],
   "attention_dropout": 0.0,
   "bos_token_id": 0,
   "decoder_attention_heads": 8,

 {
+  "_name_or_path": "ai4bharat/indictrans2-indic-indic-dist-320M",
   "activation_dropout": 0.0,
   "activation_function": "gelu",
   "architectures": [
     "IndicTransForConditionalGeneration"
   ],
+  "auto_map": {
+    "AutoConfig": "configuration_indictrans.IndicTransConfig",
+    "AutoModelForSeq2SeqLM": "modeling_indictrans.IndicTransForConditionalGeneration"
+  },
   "attention_dropout": 0.0,
   "bos_token_id": 0,
   "decoder_attention_heads": 8,