Spaces:

ameerazam08
/

Stable-Cascade-Super-Resolution

Paused

ameerazam08 commited on Feb 21, 2024

Commit

a21a448

verified ·

1 Parent(s): 6a850d5

Update modules/controlnet.py

Files changed (1) hide show

modules/controlnet.py CHANGED Viewed

@@ -229,7 +229,7 @@ class SREffnetFilter(BaseFilter):
         ])
         self.effnet = EfficientNetEncoder().to(self.device)
-        effnet_checkpoint = load_or_fail("/home/rnd/Documents/Ameer/StableCascade/models/effnet_encoder.safetensors")
         self.effnet.load_state_dict(effnet_checkpoint)
         self.effnet.eval().requires_grad_(False)

         ])
         self.effnet = EfficientNetEncoder().to(self.device)
+        effnet_checkpoint = load_or_fail("models/effnet_encoder.safetensors")
         self.effnet.load_state_dict(effnet_checkpoint)
         self.effnet.eval().requires_grad_(False)