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)
|