Update app.py

app.py

@@ -60,27 +60,27 @@ def generate(
             
         if not use_img2img:
             pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16) 
-            pipe.enable_model_cpu_offload()
+            pipe.enable_sequential_cpu_offload()
             pipe.unet.set_default_attn_processor()
             pipe.enable_vae_slicing()
 
             if use_vae:  
                 vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
                 pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
-                pipe.enable_model_cpu_offload()
+                pipe.enable_sequential_cpu_offload()
                 pipe.unet.set_default_attn_processor()
                 pipe.enable_vae_slicing()
 
         if use_img2img:
             pipe = AutoPipelineForImage2Image.from_pretrained(model, torch_dtype=torch.float16)
-            pipe.enable_model_cpu_offload()
+            pipe.enable_sequential_cpu_offload()
             pipe.unet.set_default_attn_processor()
             pipe.enable_vae_slicing()
             
             if use_vae:  
                 vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
                 pipe = AutoPipelineForImage2Image.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
-                pipe.enable_model_cpu_offload()
+                pipe.enable_sequential_cpu_offload()
                 pipe.unet.set_default_attn_processor()
                 pipe.enable_vae_slicing()