ZhengPeng7
commited on
Commit
•
282961d
1
Parent(s):
4e60c70
Add inference endpoint feature in HF model page.
Browse files- 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)
|