Modelpublic

ZhengPeng7/BiRefNet_dynamic-matting

sourceHugging Facemitupdated 8mo agoView on Hugging Face

2likes1.8kdownloads

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