all_models
/
models
/diffusers
/models--ZhengPeng7--BiRefNet
/blobs
/37c8ac58bec2f52dac34204978a7b61b69e3da76
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) | |