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