SkywalkerLu commited on
Commit
7febcc2
·
verified ·
1 Parent(s): f0e80e0

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +13 -9
config.json CHANGED
@@ -1,9 +1,13 @@
1
- {
2
- "architectures": [
3
- "TransHLA2"
4
- ],
5
- "d_model": 480,
6
- "model_type": "transhla2",
7
- "torch_dtype": "float32",
8
- "transformers_version": "4.43.4"
9
- }
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "TransHLA2"
4
+ ],
5
+ "d_model": 480,
6
+ "model_type": "transhla2",
7
+ "torch_dtype": "float32",
8
+ "transformers_version": "4.43.4",
9
+ "auto_map": {
10
+ "AutoConfig": "modeling_transhla2.TransHLA2Config",
11
+ "AutoModel": "modeling_transhla2.TransHLA2"
12
+ }
13
+ }