Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -33,11 +33,11 @@ controlnet = ControlNetModel.from_pretrained(
|
|
33 |
).to("cuda")
|
34 |
|
35 |
pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
|
36 |
-
#
|
37 |
-
"
|
38 |
controlnet = controlnet,
|
39 |
torch_dtype = torch.float16,
|
40 |
-
safety_checker =None,
|
41 |
).to("cuda")
|
42 |
#pipe.enable_xformers_memory_efficient_attention()
|
43 |
|
|
|
33 |
).to("cuda")
|
34 |
|
35 |
pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
|
36 |
+
#"digiplay/GhostMixV1.2VAE",
|
37 |
+
"CompVis/stable-diffusion-v1-4",
|
38 |
controlnet = controlnet,
|
39 |
torch_dtype = torch.float16,
|
40 |
+
safety_checker = None,
|
41 |
).to("cuda")
|
42 |
#pipe.enable_xformers_memory_efficient_attention()
|
43 |
|