Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -41,9 +41,9 @@ controlnet = ControlNetModel.from_pretrained(
|
|
| 41 |
).to("cuda")
|
| 42 |
|
| 43 |
pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
|
| 44 |
-
"digiplay/GhostMixV1.2VAE",
|
| 45 |
#"benjamin-paine/stable-diffusion-v1-5",
|
| 46 |
-
|
| 47 |
controlnet = controlnet,
|
| 48 |
torch_dtype = torch.float16,
|
| 49 |
safety_checker = None,
|
|
|
|
| 41 |
).to("cuda")
|
| 42 |
|
| 43 |
pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
|
| 44 |
+
#"digiplay/GhostMixV1.2VAE",
|
| 45 |
#"benjamin-paine/stable-diffusion-v1-5",
|
| 46 |
+
"ckpt/sd15",
|
| 47 |
controlnet = controlnet,
|
| 48 |
torch_dtype = torch.float16,
|
| 49 |
safety_checker = None,
|