idah4 commited on
Commit
c154509
·
verified ·
1 Parent(s): 9b241bc

Upload ByteETM-Korean (HF inference compatible)

Browse files
Files changed (1) hide show
  1. config.json +7 -5
config.json CHANGED
@@ -1,7 +1,5 @@
1
  {
2
- "architectures": [
3
- "HFByteETM"
4
- ],
5
  "block_size": 512,
6
  "dtype": "float32",
7
  "model_type": "byteetm",
@@ -9,5 +7,9 @@
9
  "n_head": 12,
10
  "n_layer": 4,
11
  "transformers_version": "4.57.1",
12
- "vocab_size": 258
13
- }
 
 
 
 
 
1
  {
2
+ "architectures": ["HFByteETM"],
 
 
3
  "block_size": 512,
4
  "dtype": "float32",
5
  "model_type": "byteetm",
 
7
  "n_head": 12,
8
  "n_layer": 4,
9
  "transformers_version": "4.57.1",
10
+ "vocab_size": 258,
11
+ "auto_map": {
12
+ "AutoModelForCausalLM": "modeling_byteetm.HFByteETM",
13
+ "AutoConfig": "modeling_byteetm.ByteETMConfig"
14
+ }
15
+ }