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