Update app.py

app.py

@@ -8,7 +8,7 @@ import gradio as gr
 import numpy as np
 import uuid
 import spaces
-from diffusers import ConsistencyDecoderVAE, DPMSolverMultistepScheduler, Transformer2DModel, AutoencoderKL, SASolverScheduler
+from diffusers import ConsistencyDecoderVAE, DPMSolverMultistepScheduler, Transformer2DModel, AutoencoderKL
 import torch
 from typing import Tuple
 from datetime import datetime
@@ -93,7 +93,7 @@ style_list = [
 styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
 STYLE_NAMES = list(styles.keys())
 DEFAULT_STYLE_NAME = "Realistic"
-SCHEDULE_NAME = ["DPM-Solver", "SA-Solver"]
+SCHEDULE_NAME = ["DPM-Solver"]
 DEFAULT_SCHEDULE_NAME = "DPM-Solver"
 NUM_IMAGES_PER_PROMPT = 1
 
@@ -169,9 +169,7 @@ def generate(
         height: int = 400,
         schedule: str = 'DPM-Solver',
         dpms_guidance_scale: float = 3.5,
-        sas_guidance_scale: float = 3,
         dpms_inference_steps: int = 9,
-        sas_inference_steps: int = 25,
         randomize_seed: bool = False,
         use_resolution_binning: bool = True,
         progress=gr.Progress(track_tqdm=True),
@@ -184,11 +182,6 @@ def generate(
             pipe.scheduler = DPMSolverMultistepScheduler()
         num_inference_steps = dpms_inference_steps
         guidance_scale = dpms_guidance_scale
-    elif schedule == "SA-Solver":
-        if not isinstance(pipe.scheduler, SASolverScheduler):
-            pipe.scheduler = SASolverScheduler.from_config(pipe.scheduler.config, algorithm_type='data_prediction', tau_func=lambda t: 1 if 200 <= t <= 800 else 0, predictor_order=2, corrector_order=2)
-        num_inference_steps = sas_inference_steps
-        guidance_scale = sas_guidance_scale
     else:
         raise ValueError(f"Unknown schedule: {schedule}")
 
@@ -217,16 +210,6 @@ def generate(
 
 examples = [
     "A Monkey with a happy face in the Sahara desert.",
-    "Eiffel Tower was Made up of ICE to look like a cloud, with the bell tower at the top of the building.",
-    "3D small, round, fluffy creature with big, expressive eyes explores a vibrant, enchanted forest. The creature, a whimsical blend of a rabbit and a squirrel, has soft blue fur and a bushy, striped tail. It hops along a sparkling stream, its eyes wide with wonder. The forest is alive with magical elements: flowers that glow and change colors, trees with leaves in shades of purple and silver, and small floating lights that resemble fireflies. The creature stops to interact playfully with a group of tiny, fairy-like beings dancing around a mushroom ring. The creature looks up in awe at a large, glowing tree that seems to be the heart of the forest.",
-    "Color photo of a corgi made of transparent glass, standing on the riverside in Yosemite National Park.",
-    "A close-up photo of a woman. She wore a blue coat with a gray dress underneath. She has blue eyes and blond hair, and wears a pair of earrings. Behind are blurred city buildings and streets.",
-    "A litter of golden retriever puppies playing in the snow. Their heads pop out of the snow, covered in.",
-    "a handsome young boy in the middle with sky color background wearing eye glasses, it's super detailed with anime style, it's a portrait with delicated eyes and nice looking face",
-    "an astronaut sitting in a diner, eating fries, cinematic, analog film",
-    "Pirate ship trapped in a cosmic maelstrom nebula, rendered in cosmic beach whirlpool engine, volumetric lighting, spectacular, ambient lights, intricate detail.",
-    "professional portrait photo of an anthropomorphic cat wearing fancy gentleman hat and jacket walking in autumn forest.",
-    "Outside View from Hotel Made up of Chocolate in space.",
 ]
 
 with gr.Blocks(css="style.css") as demo:
@@ -355,9 +338,7 @@ with gr.Blocks(css="style.css") as demo:
             height,
             schedule,
             dpms_guidance_scale,
-            sas_guidance_scale,
             dpms_inference_steps,
-            sas_inference_steps,
             randomize_seed,
         ],
         outputs=[result, seed],