Update BiRefNet_config.py
#6
by
kyamagu
- opened
- BiRefNet_config.py +2 -0
BiRefNet_config.py
CHANGED
|
@@ -5,7 +5,9 @@ class BiRefNetConfig(PretrainedConfig):
|
|
| 5 |
def __init__(
|
| 6 |
self,
|
| 7 |
bb_pretrained=False,
|
|
|
|
| 8 |
**kwargs
|
| 9 |
):
|
| 10 |
self.bb_pretrained = bb_pretrained
|
|
|
|
| 11 |
super().__init__(**kwargs)
|
|
|
|
| 5 |
def __init__(
|
| 6 |
self,
|
| 7 |
bb_pretrained=False,
|
| 8 |
+
is_encoder_decoder=False,
|
| 9 |
**kwargs
|
| 10 |
):
|
| 11 |
self.bb_pretrained = bb_pretrained
|
| 12 |
+
self.is_encoder_decoder = is_encoder_decoder
|
| 13 |
super().__init__(**kwargs)
|