Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -33,8 +33,8 @@ controlnet = ControlNetModel.from_pretrained(
|
|
33 |
).to("cuda")
|
34 |
|
35 |
pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
|
36 |
-
|
37 |
-
"CompVis/stable-diffusion-v1-4",
|
38 |
controlnet = controlnet,
|
39 |
torch_dtype = torch.float16,
|
40 |
safety_checker = None,
|
|
|
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,
|