from transformers import PretrainedConfig class BiRefNetConfig(PretrainedConfig): model_type = "SegformerForSemanticSegmentation" def __init__( self, bb_pretrained=False, is_encoder_decoder=False, **kwargs ): self.bb_pretrained = bb_pretrained self.is_encoder_decoder = is_encoder_decoder super().__init__(**kwargs)