Commit
•
f10d16f
1
Parent(s):
d8c0335
For users to load in one key.
Browse files- config.json +1 -4
config.json
CHANGED
|
@@ -9,7 +9,6 @@
|
|
| 9 |
},
|
| 10 |
"custom_pipelines": {
|
| 11 |
"image-segmentation": {
|
| 12 |
-
"impl": "MyPipe.RMBGPipe",
|
| 13 |
"pt": [
|
| 14 |
"AutoModelForImageSegmentation"
|
| 15 |
],
|
|
@@ -17,7 +16,5 @@
|
|
| 17 |
"type": "image"
|
| 18 |
}
|
| 19 |
},
|
| 20 |
-
"bb_pretrained": false
|
| 21 |
-
"model_type": "SegformerForSemanticSegmentation",
|
| 22 |
-
"torch_dtype": "float32",
|
| 23 |
}
|
|
|
|
| 9 |
},
|
| 10 |
"custom_pipelines": {
|
| 11 |
"image-segmentation": {
|
|
|
|
| 12 |
"pt": [
|
| 13 |
"AutoModelForImageSegmentation"
|
| 14 |
],
|
|
|
|
| 16 |
"type": "image"
|
| 17 |
}
|
| 18 |
},
|
| 19 |
+
"bb_pretrained": false
|
|
|
|
|
|
|
| 20 |
}
|