File size: 385 Bytes
7293278 6ce46b4 7293278 6ce46b4 7293278 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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)
|