reeteshmukul commited on
Commit
7f1d3b1
1 Parent(s): 1eaa0a5

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +11 -3
config.json CHANGED
@@ -1,9 +1,17 @@
1
  {
2
- "model_type" : "u2net",
3
- "image_processor_type" : "u2net",
4
  "architectures": [
5
  "U2NetForSaliencyEstimation"
6
  ],
 
 
 
 
 
 
7
  "input_channels": 3,
8
- "output_channels": 1
 
 
 
9
  }
 
1
  {
2
+ "_commit_hash": "d43d286b40c363e4dcbea1541fd4d466770e3d34",
 
3
  "architectures": [
4
  "U2NetForSaliencyEstimation"
5
  ],
6
+ "auto_map": {
7
+ "AutoConfig": "configuration_u2net.U2NetConfig",
8
+ "AutoModel": "modeling_u2net.U2NetForSaliencyEstimation"
9
+ },
10
+ "backbone_config": null,
11
+ "image_processor_type": "u2net",
12
  "input_channels": 3,
13
+ "model_type": "u2net",
14
+ "output_channels": 1,
15
+ "torch_dtype": "float32",
16
+ "transformers_version": null
17
  }