Update app.py

app.py

@@ -73,7 +73,7 @@ def generate(
     seed: int = 0,
     width: int = 1024,
     height: int = 1024,
-    prior_num_inference_steps: int = 60,
+    prior_num_inference_steps: int = 30,
     # prior_timesteps: List[float] = None,
     prior_guidance_scale: float = 4.0,
     decoder_num_inference_steps: int = 12,
@@ -90,6 +90,7 @@ def generate(
         prompt=prompt,
         height=height,
         width=width,
+        num_inference_steps=prior_num_inference_steps,
         timesteps=DEFAULT_STAGE_C_TIMESTEPS,
         negative_prompt=negative_prompt,
         guidance_scale=prior_guidance_scale,