all_models / models /BiRefNet /RMBG-2.0 /BiRefNet_config.py
jaxmetaverse's picture
Upload folder using huggingface_hub
5ca481c 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)