ummagumm-a commited on
Commit
a811984
·
1 Parent(s): 7850b8b

Upload model

Browse files
Files changed (2) hide show
  1. config.json +3 -2
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,11 +1,12 @@
1
  {
2
  "architectures": [
3
- "AirEmbedder"
4
  ],
 
5
  "dropout_p": 0.01,
6
  "hidden_dim": 50,
7
  "input_dim": 371,
8
- "model_type": "mymodel",
9
  "num_classes": 6,
10
  "num_layers": 1,
11
  "torch_dtype": "float32",
 
1
  {
2
  "architectures": [
3
+ "AirModelHF"
4
  ],
5
+ "device": "cpu",
6
  "dropout_p": 0.01,
7
  "hidden_dim": 50,
8
  "input_dim": 371,
9
+ "model_type": "gru",
10
  "num_classes": 6,
11
  "num_layers": 1,
12
  "torch_dtype": "float32",
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c86b8cd2b8a06743d1157975af73f53f7c3f3f7dea9c78c5252ff30fc172c846
3
- size 343259621
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c413caa62ad75c0f9985bfb3a8921278a16c388dec27853fb95bb54bacfe5520
3
+ size 513039