Datasetpublic

maoper11/RMBG-2.0

sourceHugging Faceupdated 4mo agoView on Hugging Face

0likes9downloads

BiRefNet_config.py12 linesDownload Raw Back to root

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