add inference

app.py

@@ -176,7 +176,11 @@ accelerator = accelerate.Accelerator()
 def generate_image_with_steps(prompt, negative_prompt, seed, width, height, guidance_scale, num_inference_steps):
     """Helper function to generate image with specific number of steps"""
     prompts =  [prompt]
-    sampler = UniPCSampler(pipe,model_closure=model_closure, steps=num_inference_steps, guidance_scale=guidance_scale)
+    if num_inference_steps > 8:
+        sampler = UniPCSampler(pipe,model_closure=model_closure, steps=num_inference_steps, guidance_scale=guidance_scale,skip_type='time_uniform')
+    else:
+        sampler = UniPCSampler(pipe,model_closure=model_closure, steps=num_inference_steps, guidance_scale=guidance_scale)
+
     c = prompts
     uc = ['(worst quality:2), (low quality:2), (normal quality:2), bad anatomy, bad proportions, poorly drawn face, poorly drawn hands, missing fingers, extra limbs, blurry, pixelated, distorted, lowres, jpeg artifacts, watermark, signature, text, (deformed:1.5), (bad hands:1.3), overexposed, underexposed, censored, mutated, extra fingers, cloned face, bad eyes'] * len(c) if guidance_scale != 1.0 else None
     shape = [4, width // 8, height // 8]
@@ -226,7 +230,7 @@ def infer(
     image_quick = generate_image_with_steps(prompt, negative_prompt, seed, width, height, guidance_scale, num_inference_steps)
     
     # Generate image with 50 steps for high quality
-    image_50_steps = generate_image_with_steps(prompt, negative_prompt, seed, width, height, guidance_scale, 50)
+    image_50_steps = generate_image_with_steps(prompt, negative_prompt, seed, width, height, guidance_scale, 20)
 
     return image_quick, image_50_steps, seed
 
@@ -262,11 +266,11 @@ with gr.Blocks() as demo:
 
         with gr.Row():
             with gr.Column():
-                gr.Markdown("### Our fast inference Result")
+                gr.Markdown("### Our fast inference Result using afs and uni-predictor to get 2 free steps")
                 result = gr.Image(label="Quick Result", show_label=False)
             with gr.Column():
-                gr.Markdown("### Original 50 steps Result")
-                result_50_steps = gr.Image(label="50 Steps Result", show_label=False)
+                gr.Markdown("### Original 20 steps Result")
+                result_20_steps = gr.Image(label="20 Steps Result", show_label=False)
 
         with gr.Accordion("Advanced Settings", open=False):
             negative_prompt = gr.Text(