Spaces:

Baptlem
/

UCDR-Net

Sleeping

Baptlem commited on May 2, 2023

Commit

0a95bff

•

1 Parent(s): 6f70200

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ def load_sb_pipe(controlnet_version, sb_path="runwayml/stable-diffusion-v1-5"):
     controlnet, controlnet_params = load_controlnet(controlnet_version)
     scheduler, scheduler_params = FlaxDPMSolverMultistepScheduler.from_pretrained(
-        base_model_path,
         subfolder="scheduler"
     )

     controlnet, controlnet_params = load_controlnet(controlnet_version)
     scheduler, scheduler_params = FlaxDPMSolverMultistepScheduler.from_pretrained(
+        sb_path,
         subfolder="scheduler"
     )