add app.py

app.py

@@ -4,18 +4,7 @@ import torch
 from PIL import Image
 import os
 
-scheduler = UniPCMultistepScheduler(
-    beta_start=0.00085,
-    beta_end=0.012,
-    beta_schedule="scaled_linear",
-    solver_order=2,
-    prediction_type="epsilon",
-    predict_x0=True,
-    thresholding=False,
-    solver_type='bh2',
-    lower_order_final=True,
-    disable_corrector=[0],
-)
+
 
 class Model:
     def __init__(self, name, path, prefix):
@@ -89,6 +78,7 @@ def txt_to_img(model_path, prompt, neg_prompt, guidance, steps, width, height, g
 
         pipe.to("cpu")
         pipe = current_model.pipe_t2i
+        pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 
         if torch.cuda.is_available():
           pipe = pipe.to("cuda")
@@ -117,6 +107,7 @@ def img_to_img(model_path, prompt, neg_prompt, img, strength, guidance, steps, w
 
         pipe.to("cpu")
         pipe = current_model.pipe_i2i
+        pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
         
         if torch.cuda.is_available():
           pipe = pipe.to("cuda")