mirror of
https://www.modelscope.cn/AI-ModelScope/RMBG-2.0.git
synced 2025-04-10 11:50:22 +08:00
12 lines
298 B
Python
12 lines
298 B
Python
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)
|