Update app.py

app.py

@@ -37,7 +37,7 @@ controlnet = ControlNetModel.from_pretrained(
     "monster-labs/control_v1p_sd15_qrcode_monster",
     #"DionTimmer/controlnet_qrcode-control_v1p_sd15", 
     torch_dtype=torch.float16
-).to("cpu")
+).to("cuda")
 
 pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
     "digiplay/GhostMixV1.2VAE",
@@ -45,8 +45,10 @@ pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
     controlnet = controlnet,
     torch_dtype = torch.float16,
     safety_checker = None,
-).to("cpu")
-#pipe.enable_xformers_memory_efficient_attention()
+).to("cuda")
+pipe.enable_xformers_memory_efficient_attention()
+pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
+pipe.enable_model_cpu_offload()
 
 
 def resize_for_condition_image(input_image: Image.Image, resolution: int):