yairschiff commited on
Commit
4c437fc
·
verified ·
1 Parent(s): e07ddf4

Initial commit

Browse files
Files changed (2) hide show
  1. config.json +4 -0
  2. pytorch_model.bin +1 -1
config.json CHANGED
@@ -2,6 +2,10 @@
2
  "architectures": [
3
  "MyModel"
4
  ],
 
 
 
 
5
  "hidden_dim": 128,
6
  "input_dim": 2,
7
  "model_type": "my_model",
 
2
  "architectures": [
3
  "MyModel"
4
  ],
5
+ "auto_map": {
6
+ "AutoConfig": "modeling.MyModelConfig",
7
+ "AutoModel": "modeling.MyModel"
8
+ },
9
  "hidden_dim": 128,
10
  "input_dim": 2,
11
  "model_type": "my_model",
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:26b9155a5ae09cb107090ab90f216cee146012b9c259624b4f2d3b4605b0fd0d
3
  size 137864
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de5d627b2857a897b505891be2fa786a0c496def721c17738b4bcb658f634379
3
  size 137864