ameerazam08 commited on
Commit
a21a448
·
verified ·
1 Parent(s): 6a850d5

Update modules/controlnet.py

Browse files
Files changed (1) hide show
  1. 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("/home/rnd/Documents/Ameer/StableCascade/models/effnet_encoder.safetensors")
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