Severian commited on
Commit
ea06e6e
·
verified ·
1 Parent(s): 4ea22d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -33,11 +33,11 @@ controlnet = ControlNetModel.from_pretrained(
33
  ).to("cuda")
34
 
35
  pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
36
- # "runwayml/stable-diffusion-v1-5",
37
- "digiplay/GhostMixV1.2VAE",
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