giga-lora-studio-models / BiRefNet /BiRefNet_config.py
ChrisColeTech's picture
Copy files from models/1038lab/BiRefNet
ae2ccd5 verified
Raw History Blame Contribute Delete
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)