Hugging Face's logo

nz
/
RITA_m

Feature Extraction
Transformers
PyTorch
codegen
custom_code
Model card Files Community
nz commited on
Commit
5c630e6
•
1 Parent(s): b6368ec

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -11
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "Seledorn/RITA_m",
3
  "architectures": [
4
  "RITAModel"
5
  ],
@@ -7,22 +7,15 @@
7
  "AutoConfig": "rita_configuration.RITAConfig",
8
  "AutoModel": "rita_modeling.RITAModel"
9
  },
10
- "bos_token_id": [
11
- [
12
- [
13
- 50256
14
- ]
15
- ]
16
- ],
17
  "d_feedforward": 4096,
18
  "d_model": 1024,
19
  "dropout": 0.0,
20
- "eos_token_id": 50256,
21
  "max_seq_len": 1024,
22
- "model_type": "codegen",
23
  "num_heads": 16,
24
  "num_layers": 24,
25
  "torch_dtype": "float16",
26
  "transformers_version": "4.18.0",
27
- "vocab_size": 128
28
  }
 
1
  {
2
+ "_name_or_path": "nz/RITA_m",
3
  "architectures": [
4
  "RITAModel"
5
  ],
 
7
  "AutoConfig": "rita_configuration.RITAConfig",
8
  "AutoModel": "rita_modeling.RITAModel"
9
  },
 
 
 
 
 
 
 
10
  "d_feedforward": 4096,
11
  "d_model": 1024,
12
  "dropout": 0.0,
13
+ "eos_token_id": 2,
14
  "max_seq_len": 1024,
15
+ "model_type": "rita",
16
  "num_heads": 16,
17
  "num_layers": 24,
18
  "torch_dtype": "float16",
19
  "transformers_version": "4.18.0",
20
+ "vocab_size": 26
21
  }