Severian commited on
Commit
b106bc6
·
verified ·
1 Parent(s): 706a298

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -37,7 +37,7 @@ controlnet = ControlNetModel.from_pretrained(
37
  "monster-labs/control_v1p_sd15_qrcode_monster",
38
  #"DionTimmer/controlnet_qrcode-control_v1p_sd15",
39
  torch_dtype=torch.float16
40
- ).to("mps")
41
 
42
  pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
43
  "digiplay/GhostMixV1.2VAE",
@@ -45,7 +45,7 @@ pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
45
  controlnet = controlnet,
46
  torch_dtype = torch.float16,
47
  safety_checker = None,
48
- ).to("mps")
49
  #pipe.enable_xformers_memory_efficient_attention()
50
 
51
 
 
37
  "monster-labs/control_v1p_sd15_qrcode_monster",
38
  #"DionTimmer/controlnet_qrcode-control_v1p_sd15",
39
  torch_dtype=torch.float16
40
+ ).to("cuda")
41
 
42
  pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
43
  "digiplay/GhostMixV1.2VAE",
 
45
  controlnet = controlnet,
46
  torch_dtype = torch.float16,
47
  safety_checker = None,
48
+ ).to("cuda")
49
  #pipe.enable_xformers_memory_efficient_attention()
50
 
51