Spaces:

Underground-Digital
/

Code-Cooker

Build error

Severian commited on Sep 6, 2024

Commit

ea06e6e

verified ·

1 Parent(s): 4ea22d7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,11 +33,11 @@ controlnet = ControlNetModel.from_pretrained(
 ).to("cuda")
 pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
-    # "runwayml/stable-diffusion-v1-5",
-    "digiplay/GhostMixV1.2VAE",
     controlnet = controlnet,
     torch_dtype = torch.float16,
-    safety_checker =None,
 ).to("cuda")
 #pipe.enable_xformers_memory_efficient_attention()

 ).to("cuda")
 pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
+    #"digiplay/GhostMixV1.2VAE",
+    "CompVis/stable-diffusion-v1-4",
     controlnet = controlnet,
     torch_dtype = torch.float16,
+    safety_checker = None,
 ).to("cuda")
 #pipe.enable_xformers_memory_efficient_attention()