Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import numpy as np
 import spaces
 import torch
-from diffusers import AutoPipelineForText2Image, EulerAncestralDiscreteScheduler
+from diffusers import AutoPipelineForText2Image, EulerDiscreteScheduler
 
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>你现在运行在CPU上 但是只支持GPU.</p>"
@@ -18,20 +18,7 @@ if torch.cuda.is_available():
         use_safetensors=True,
         add_watermarker=False
     )
-    pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-    pipe.scheduler.register_to_config(
-        rescale_betas_zero_snr=True,
-        clip_sample=False,
-        interpolation_type=linear,
-        prediction_type=epsilon,
-        sample_max_value=1.0,
-        set_alpha_to_one=False,
-        skip_prk_steps=True,
-        steps_offset=1,
-        timestep_spacing=leading,
-        trained_betas=null,
-        use_karras_sigmas=False
-    )
+    pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", prediction_type="sample" if num_inference_steps==1 else "epsilon")
     pipe.to("cuda")
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int: