jaygala24 commited on
Commit
9543668
1 Parent(s): 80e03c7

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +5 -1
config.json CHANGED
@@ -1,10 +1,14 @@
1
  {
2
- "_name_or_path": "/data/jaygala/it2_ckpts/hf_base_models/indic-indic",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "architectures": [
6
  "IndicTransForConditionalGeneration"
7
  ],
 
 
 
 
8
  "attention_dropout": 0.0,
9
  "bos_token_id": 0,
10
  "decoder_attention_heads": 16,
 
1
  {
2
+ "_name_or_path": "ai4bharat/indictrans2-indic-indic-1B",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "architectures": [
6
  "IndicTransForConditionalGeneration"
7
  ],
8
+ "auto_map": {
9
+ "AutoConfig": "configuration_indictrans.IndicTransConfig",
10
+ "AutoModelForSeq2SeqLM": "modeling_indictrans.IndicTransForConditionalGeneration"
11
+ },
12
  "attention_dropout": 0.0,
13
  "bos_token_id": 0,
14
  "decoder_attention_heads": 16,