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