Spaces:

shichen1231
/

illumination_composition_controlnet

Runtime error

shichen1231 commited on Jun 7, 2023

Commit

0525e06

1 Parent(s): f2467d5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,12 +6,12 @@ torch.backends.cuda.matmul.allow_tf32 = True
 import gc
 model_id = "andite/pastel-mix"
 controlnet = ControlNetModel.from_pretrained("ioclab/connow", torch_dtype=torch.float16, use_safetensors=True)
-pipe = StableDiffusionControlNetPipeline.from_pretrained(
-    model_id,
     controlnet=controlnet,
     torch_dtype=torch.float16,
     safety_checker=None,

 import gc
 model_id = "andite/pastel-mix"
+model_url = "https://huggingface.co/andite/pastel-mix/blob/main/pastelmix-better-vae.ckpt"
 controlnet = ControlNetModel.from_pretrained("ioclab/connow", torch_dtype=torch.float16, use_safetensors=True)
+pipe = StableDiffusionControlNetPipeline.from_from_ckpt(
+    model_url,
     controlnet=controlnet,
     torch_dtype=torch.float16,
     safety_checker=None,