Make some changes to try and fix out of memory issue.

app.py

@@ -38,7 +38,9 @@ pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
     safety_checker=None,
     torch_dtype=torch.float16,
 ).to("cuda")
-# pipe.enable_xformers_memory_efficient_attention()
+pipe.enable_xformers_memory_efficient_attention()
+pipe.controlnet = torch.nn.DataParallel(pipe.controlnet)
+pipe.unet = torch.nn.DataParallel(pipe.unet)
 
 
 def resize_for_condition_image(input_image: Image.Image, resolution: int, canvas_width: int = None, canvas_height: int = None):
@@ -319,5 +321,6 @@ model: https://huggingface.co/DionTimmer/controlnet_qrcode-control_v1p_sd15
         cache_examples=True,
     )
 
+os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
 blocks.queue(max_size=20,api_open=False)
 blocks.launch(share=bool(os.environ.get("SHARE", False)), show_api=False)