BiRefNet / BiRefNet_config.py
ZhengPeng7's picture
Ensemble BiRefNet codes to HF to load model with weights in one line.
40a7dcf
raw
history blame
298 Bytes
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)