not-lain commited on
Commit
5c5209f
1 Parent(s): 827dc1a

Upload MnistPipe

Browse files
Files changed (1) hide show
  1. config.json +14 -3
config.json CHANGED
@@ -1,14 +1,25 @@
1
  {
 
2
  "architectures": [
3
  "MnistModel"
4
  ],
5
  "auto_map": {
6
- "AutoConfig": "MyConfig.MnistConfig",
7
- "AutoModelForImageClassification": "MyModel.MnistModel"
8
  },
9
  "conv1": 10,
10
  "conv2": 20,
 
 
 
 
 
 
 
 
 
 
11
  "model_type": "MobileNetV1",
12
  "torch_dtype": "float32",
13
- "transformers_version": "4.38.2"
14
  }
 
1
  {
2
+ "_name_or_path": "not-lain/MyRepo",
3
  "architectures": [
4
  "MnistModel"
5
  ],
6
  "auto_map": {
7
+ "AutoConfig": "not-lain/MyRepo--MyConfig.MnistConfig",
8
+ "AutoModelForImageClassification": "not-lain/MyRepo--MyModel.MnistModel"
9
  },
10
  "conv1": 10,
11
  "conv2": 20,
12
+ "custom_pipelines": {
13
+ "image-classification": {
14
+ "impl": "MyPipe.MnistPipe",
15
+ "pt": [
16
+ "AutoModelForImageClassification"
17
+ ],
18
+ "tf": [],
19
+ "type": "image"
20
+ }
21
+ },
22
  "model_type": "MobileNetV1",
23
  "torch_dtype": "float32",
24
+ "transformers_version": "4.40.0"
25
  }