Modelpublic

1038lab/BiRefNet

sourceHugging Faceupdated 2mo agoView on Hugging Face

9likes20kdownloads

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