tai-dang11 commited on
Commit
3e41eba
·
1 Parent(s): 89a696a
my_model/config.json → config.json RENAMED
@@ -1,10 +1,8 @@
1
  {
2
  "architectures": [
3
- "CustomModel"
4
  ],
5
- "hidden_size": 512,
6
- "model_type": "custom_model",
7
- "num_classes": 10,
8
  "torch_dtype": "float32",
9
  "transformers_version": "4.42.4"
10
  }
 
1
  {
2
  "architectures": [
3
+ "SimpleCustomModel"
4
  ],
5
+ "model_type": "simple_custom",
 
 
6
  "torch_dtype": "float32",
7
  "transformers_version": "4.42.4"
8
  }
my_model/model.safetensors → model.safetensors RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:741bc8a7e3aa3e39feea9c1fd7e4f7a4c9c54d3a44f105043592bf0d7252e9dc
3
- size 20704
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbec86fec34981fc2a09884cd9730e476fd640a7ba81484fe996fc589b95d8be
3
+ size 6336