Update app.py

app.py

@@ -57,28 +57,26 @@ def generate(
     url = '',
 ):
     if torch.cuda.is_available():
-    
+
+        pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+        
         if not use_img2img:
             pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
             pipe.enable_model_cpu_offload()
-            pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
             
             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.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
         if use_img2img:
             pipe = AutoPipelineForImage2Image.from_pretrained(model, torch_dtype=torch.float16)
             pipe.enable_model_cpu_offload()
-            pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
             
             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.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
             response = requests.get(url)
             init_image = Image.open(BytesIO(response.content)).convert("RGB")