ZhengPeng7 commited on
Commit
282961d
1 Parent(s): 4e60c70

Add inference endpoint feature in HF model page.

Browse files
Files changed (1) hide show
  1. birefnet.py +2 -1
birefnet.py CHANGED
@@ -1995,7 +1995,8 @@ class BiRefNet(
1995
  ):
1996
  config_class = BiRefNetConfig
1997
  def __init__(self, bb_pretrained=True, config=BiRefNetConfig()):
1998
- super(BiRefNet, self).__init__()
 
1999
  self.config = Config()
2000
  self.epoch = 1
2001
  self.bb = build_backbone(self.config.bb, pretrained=bb_pretrained)
 
1995
  ):
1996
  config_class = BiRefNetConfig
1997
  def __init__(self, bb_pretrained=True, config=BiRefNetConfig()):
1998
+ super(BiRefNet, self).__init__(config)
1999
+ bb_pretrained = config.bb_pretrained
2000
  self.config = Config()
2001
  self.epoch = 1
2002
  self.bb = build_backbone(self.config.bb, pretrained=bb_pretrained)