AlanRobotics commited on
Commit
7b8d8df
·
1 Parent(s): cf14450

Upload model

Browse files
Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -3,7 +3,8 @@
3
  "SiamseNNModel"
4
  ],
5
  "auto_map": {
6
- "AutoModel": "configuration_siamese.SiameseConfig"
 
7
  },
8
  "model_type": "AutoModel",
9
  "torch_dtype": "float32",
 
3
  "SiamseNNModel"
4
  ],
5
  "auto_map": {
6
+ "AutoConfig": "configuration_siamese.SiameseConfig",
7
+ "AutoModel": "modeling_siamese.SiamseNNModel"
8
  },
9
  "model_type": "AutoModel",
10
  "torch_dtype": "float32",