Update app.py

app.py

@@ -7,7 +7,12 @@ from multiprocessing import cpu_count
 import requests
 import io
 import os
+
+
 from PIL import Image
+from diffusers import StableDiffusionControlNetImg2ImgPipeline, ControlNetModel, DDIMScheduler
+from diffusers.utils import load_image
+
 
 from diffusers import (
     StableDiffusionPipeline,
@@ -38,6 +43,8 @@ pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
     torch_dtype=torch.float16,
 ).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):