Code-Cooker

Sleeping

Severian commited on Sep 23, 2024

Commit

05785e6

verified ·

1 Parent(s): d8ecf0f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -70,7 +70,7 @@ def load_models_on_launch():
         torch_dtype=torch.float16
     ).to("cuda")
-    diffusion_path = snapshot_download(DIFFUSION_MODELS["GhostMix", "Stable v1.5"])
     loaded_pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
         diffusion_path,
         controlnet=loaded_controlnet,

         torch_dtype=torch.float16
     ).to("cuda")
+    diffusion_path = snapshot_download(DIFFUSION_MODELS["GhostMix"])
     loaded_pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
         diffusion_path,
         controlnet=loaded_controlnet,