Upload folder using huggingface_hub
Browse files- birefnet.py +1 -3
birefnet.py
CHANGED
|
@@ -2065,9 +2065,7 @@ class BiRefNet(
|
|
| 2065 |
if 'bb.' in key and 'refiner.' not in key:
|
| 2066 |
value.requires_grad = False
|
| 2067 |
|
| 2068 |
-
|
| 2069 |
-
self.post_init()
|
| 2070 |
-
|
| 2071 |
|
| 2072 |
def forward_enc(self, x):
|
| 2073 |
if self.config.bb in ['vgg16', 'vgg16bn', 'resnet50']:
|
|
|
|
| 2065 |
if 'bb.' in key and 'refiner.' not in key:
|
| 2066 |
value.requires_grad = False
|
| 2067 |
|
| 2068 |
+
self.post_init()
|
|
|
|
|
|
|
| 2069 |
|
| 2070 |
def forward_enc(self, x):
|
| 2071 |
if self.config.bb in ['vgg16', 'vgg16bn', 'resnet50']:
|