Severian commited on
Commit
95b8304
·
verified ·
1 Parent(s): 1855188

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -41,8 +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
  controlnet = controlnet,
47
  torch_dtype = torch.float16,
48
  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,