maxoul commited on
Commit
071f001
·
verified ·
1 Parent(s): 659af08

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +5 -1
config.json CHANGED
@@ -14,5 +14,9 @@
14
  "padding_side": "left",
15
  "torch_dtype": "bfloat16",
16
  "train_head": false,
17
- "transformers_version": "4.53.3"
 
 
 
 
18
  }
 
14
  "padding_side": "left",
15
  "torch_dtype": "bfloat16",
16
  "train_head": false,
17
+ "transformers_version": "4.53.3",
18
+ "auto_map": {
19
+ "AutoConfig": "splade.SpladeConfig",
20
+ "AutoModelForCausalLM": "splade.Splade"
21
+ }
22
  }