| 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) | |
| 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) | |