ameerazam08
commited on
Update modules/controlnet.py
Browse files- modules/controlnet.py +1 -1
modules/controlnet.py
CHANGED
@@ -229,7 +229,7 @@ class SREffnetFilter(BaseFilter):
|
|
229 |
])
|
230 |
|
231 |
self.effnet = EfficientNetEncoder().to(self.device)
|
232 |
-
effnet_checkpoint = load_or_fail("
|
233 |
self.effnet.load_state_dict(effnet_checkpoint)
|
234 |
self.effnet.eval().requires_grad_(False)
|
235 |
|
|
|
229 |
])
|
230 |
|
231 |
self.effnet = EfficientNetEncoder().to(self.device)
|
232 |
+
effnet_checkpoint = load_or_fail("models/effnet_encoder.safetensors")
|
233 |
self.effnet.load_state_dict(effnet_checkpoint)
|
234 |
self.effnet.eval().requires_grad_(False)
|
235 |
|