Modelpublic

yamura4/RMBG-2.0-WebGPU

sourceHugging Facecc-by-nc-4.0updated 3mo agoView on Hugging Face

4likes26downloads

BiRefNet_config.py12 linesDownload Raw Back to root

1from transformers import PretrainedConfig2 3class BiRefNetConfig(PretrainedConfig):4    model_type = "birefnet"5    def __init__(6        self,7        bb_pretrained=False,8        **kwargs9    ):10        self.bb_pretrained = bb_pretrained11        super().__init__(**kwargs)12