Update app.py

app.py

@@ -261,8 +261,12 @@ def generate(
     finally:
         if use_upscaler:
             del upscaler_pipe
-        pipe.scheduler = backup_scheduler
-        img_pipe.scheduler = img_backup_scheduler
+
+        if isImg2Img:
+            img_pipe.scheduler = img_backup_scheduler
+        else:
+            pipe.scheduler = backup_scheduler
+        
         utils.free_memory()
 
 
@@ -420,7 +424,7 @@ with gr.Blocks(css="style.css", theme="NoCrypt/[email protected]") as demo:
                 examples=config.examples,
                 inputs=prompt,
                 outputs=[result, gr_metadata],
-                fn=lambda *args, **kwargs: generate(*args, use_upscaler=True, **kwargs),
+                #fn=lambda *args, **kwargs: generate(*args, use_upscaler=True, **kwargs),
                 cache_examples=CACHE_EXAMPLES,
             )
     use_upscaler.change(