BiRefNet / BiRefNet_config.py
ZhengPeng7's picture
For users to load in one key.
fc65c0f
raw
history blame
No virus
224 Bytes
from transformers import PretrainedConfig
class BiRefNetConfig(PretrainedConfig):
model_type = "SegformerForSemanticSegmentation"
def __init__(
self,
**kwargs
):
super().__init__(**kwargs)