razmars commited on
Commit
fe777ca
·
verified ·
1 Parent(s): 84113e3

Update zero_shot/config.json

Browse files
Files changed (1) hide show
  1. zero_shot/config.json +5 -1
zero_shot/config.json CHANGED
@@ -1,7 +1,11 @@
1
- { "_name_or_path": "super_linear",
 
2
  "architectures": [
3
  "SuperLinearForCausalLM"
4
  ],
 
 
 
5
  "auto_regressive": 1,
6
  "d_model": 128,
7
  "dropout": 0.0,
 
1
+ {
2
+ "_name_or_path": "super_linear",
3
  "architectures": [
4
  "SuperLinearForCausalLM"
5
  ],
6
+ "auto_map": {
7
+ "AutoModelForCausalLM": "modeling_super_linear.SuperLinearForCausalLM"
8
+ },
9
  "auto_regressive": 1,
10
  "d_model": 128,
11
  "dropout": 0.0,