Update config.json
Browse files- config.json +12 -7
config.json
CHANGED
|
@@ -11,6 +11,18 @@
|
|
| 11 |
"label2id": {
|
| 12 |
"object": 0
|
| 13 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
"mask_decoder_config": {
|
| 15 |
"_name_or_path": "",
|
| 16 |
"add_cross_attention": false,
|
|
@@ -85,13 +97,6 @@
|
|
| 85 |
"typical_p": 1.0,
|
| 86 |
"use_bfloat16": false
|
| 87 |
},
|
| 88 |
-
"model_type": "sam",
|
| 89 |
-
"pipeline_tag": "image-segmentation",
|
| 90 |
-
"task_specific_params": {
|
| 91 |
-
"image-segmentation": {
|
| 92 |
-
"checkpoint": "sam-vit-base"
|
| 93 |
-
}
|
| 94 |
-
},
|
| 95 |
"prompt_encoder_config": {
|
| 96 |
"_name_or_path": "",
|
| 97 |
"add_cross_attention": false,
|
|
|
|
| 11 |
"label2id": {
|
| 12 |
"object": 0
|
| 13 |
},
|
| 14 |
+
"model_type": "sam",
|
| 15 |
+
"pipeline_tag": "image-segmentation",
|
| 16 |
+
"subtask": "segmentation",
|
| 17 |
+
"default_subtask": "segmentation",
|
| 18 |
+
"task_specific_params": {
|
| 19 |
+
"image-segmentation": {
|
| 20 |
+
"checkpoint": "sam-vit-base",
|
| 21 |
+
"subtask": "segmentation",
|
| 22 |
+
"input_type": "points",
|
| 23 |
+
"output_type": "masks"
|
| 24 |
+
}
|
| 25 |
+
},
|
| 26 |
"mask_decoder_config": {
|
| 27 |
"_name_or_path": "",
|
| 28 |
"add_cross_attention": false,
|
|
|
|
| 97 |
"typical_p": 1.0,
|
| 98 |
"use_bfloat16": false
|
| 99 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 100 |
"prompt_encoder_config": {
|
| 101 |
"_name_or_path": "",
|
| 102 |
"add_cross_attention": false,
|