Update app.py

app.py

@@ -24,6 +24,7 @@ import paramiko
 import gc
 import time
 import datetime
+from diffusers.schedulers import AysSchedules
 
 #os.system("chmod +x ./cusparselt.sh")
 #os.system("./cusparselt.sh")
@@ -97,7 +98,8 @@ DEFAULT_STYLE_NAME = "Style Zero"
 STYLE_NAMES = list(styles.keys())
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-   
+sampling_schedule = AysSchedules["StableDiffusionXLTimesteps"]
+
 def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
     if style_name in styles:
         p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
@@ -118,9 +120,9 @@ def load_and_prepare_model(model_id):
   #  vae = AutoencoderKL.from_pretrained("BeastHF/MyBack_SDXL_Juggernaut_XL_VAE/MyBack_SDXL_Juggernaut_XL_VAE_V10(version_X).safetensors",safety_checker=None).to(torch.bfloat16)
     #  vae = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", safety_checker=None).to('cuda')
     #sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/Juggernaut-XI-v11-fp32', subfolder='scheduler',beta_schedule="scaled_linear",use_karras_sigmas=True)
-    #sched = EulerAncestralDiscreteScheduler.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='scheduler',beta_schedule="scaled_linear", steps_offset=1)
-    #sched = EulerAncestralDiscreteScheduler()
-    sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1,use_karras_sigmas=True)
+    #sched = EulerAncestralDiscreteScheduler.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='scheduler',beta_schedule="scaled_linear", steps_offset=1,timestep_spacing="trailing"))
+    sched = EulerAncestralDiscreteScheduler(timestep_spacing="trailing",steps_offset=1)
+    #sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1,use_karras_sigmas=True)
       #  sched = EulerAncestralDiscreteScheduler.from_config('ford442/RealVisXL_V5.0_BF16', beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1)
     pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0")
     
@@ -237,6 +239,7 @@ def generate_30(
         "guidance_scale": guidance_scale,
         "num_inference_steps": num_inference_steps,
         "generator": generator,
+        "timesteps": sampling_schedule,
         "output_type": "pil",
     }
     if use_resolution_binning:
@@ -303,6 +306,7 @@ def generate_60(
         "guidance_scale": guidance_scale,
         "num_inference_steps": num_inference_steps,
         "generator": generator,
+        "timesteps": sampling_schedule,
         "output_type": "pil",
     }
     if use_resolution_binning:
@@ -369,6 +373,7 @@ def generate_90(
         "guidance_scale": guidance_scale,
         "num_inference_steps": num_inference_steps,
         "generator": generator,
+        "timesteps": sampling_schedule,
         "output_type": "pil",
     }
     if use_resolution_binning: