DanielHesslow commited on
Commit
e283837
1 Parent(s): 7f52a15
Files changed (1) hide show
  1. config.json +6 -3
config.json CHANGED
@@ -1,16 +1,19 @@
1
  {
2
- "_name_or_path": "./test_model",
3
  "architectures": [
4
  "RITAModel"
5
  ],
6
  "auto_map": {
7
  "AutoConfig": "rita_configuration.RITAConfig",
8
- "AutoModel": "rita_modeling.RITAModel"
 
9
  },
10
  "bos_token_id": [
11
  [
12
  [
13
- 50256
 
 
14
  ]
15
  ]
16
  ],
 
1
  {
2
+ "_name_or_path": "Seledorn/RITA_s",
3
  "architectures": [
4
  "RITAModel"
5
  ],
6
  "auto_map": {
7
  "AutoConfig": "rita_configuration.RITAConfig",
8
+ "AutoModel": "rita_modeling.RITAModel",
9
+ "AutoModelForCausalLM": "rita_modeling.RITAModel"
10
  },
11
  "bos_token_id": [
12
  [
13
  [
14
+ [
15
+ 50256
16
+ ]
17
  ]
18
  ]
19
  ],