Update BiRefNet_config.py

#6
by kyamagu - opened
Files changed (1) hide show
  1. 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)