Modelpublic

not-lain/CustomCodeForRMBG

sourceHugging Faceupdated 3y agoView on Hugging Face

0likes8downloads

config.json26 linesDownload Raw Back to root

1{2  "_name_or_path": "not-lain/CustomCodeForRMBG",3  "architectures": [4    "BriaRMBG"5  ],6  "auto_map": {7    "AutoConfig": "not-lain/CustomCodeForRMBG--MyConfig.RMBGConfig",8    "AutoModelForImageSegmentation": "not-lain/CustomCodeForRMBG--briarmbg.BriaRMBG"9  },10  "custom_pipelines": {11    "image-segmentation": {12      "impl": "MyPipe.RMBGPipe",13      "pt": [14        "AutoModelForImageSegmentation"15      ],16      "tf": [],17      "type": "image"18    }19  },20  "in_ch": 3,21  "model_type": "SegformerForSemanticSegmentation",22  "out_ch": 1,23  "torch_dtype": "float32",24  "transformers_version": "4.38.0.dev0"25}26