Update app.py

app.py

@@ -63,6 +63,9 @@ def infer(
     system_prompt=default_system_prompt,
     cfg_normalization=True,
     cfg_trunc_ratio=1.0,
+    max_sequence_length=256,
+    dynamic_shifting=True,
+    sigmas="Default",
     progress=gr.Progress(track_tqdm=True),
 ):
     if randomize_seed:
@@ -70,6 +73,11 @@ def infer(
 
     generator = torch.Generator().manual_seed(seed)
 
+    pipe.scheduler.config.use_dynamic_shifting = dynamic_shifting
+    pipe.scheduler.config.use_karras_sigmas = sigmas == "Karras"
+    pipe.scheduler.config.use_exponential_sigmas = sigmas == "Exponential"
+    pipe.scheduler.config.use_beta_sigmas = sigmas == "Beta"
+    
     image = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -79,8 +87,9 @@ def infer(
         height=height,
         generator=generator,
         system_prompt=system_prompt,
+        max_sequence_length=max_sequence_length,
         cfg_normalization=cfg_normalization,
-        cfg_trunc_ratio=cfg_trunc_ratio
+        cfg_trunc_ratio=cfg_trunc_ratio,
     ).images[0]
 
     return image, seed
@@ -104,6 +113,7 @@ with gr.Blocks(css=css) as demo:
             prompt = gr.Text(
                 label="Prompt",
                 show_label=False,
+                lines=2,
                 max_lines=4,
                 placeholder="Enter your prompt",
                 container=False,
@@ -117,12 +127,14 @@ with gr.Blocks(css=css) as demo:
             with gr.Row():
                 system_prompt = gr.Text(
                     label="System Prompt",
+                    lines=2,
                     max_lines=4,
                     value=default_system_prompt
                 )
                 
                 negative_prompt = gr.Text(
                     label="Negative prompt",
+                    lines=2,
                     max_lines=4,
                     placeholder="Enter a negative prompt",
                 )
@@ -172,6 +184,14 @@ with gr.Blocks(css=css) as demo:
                     value=30, 
                 )
 
+                max_sequence_length = gr.Slider(
+                    label="Max Sequence Length",
+                    minimum=16,
+                    maximum=512,
+                    value=256,
+                    step=8
+                )
+                
             with gr.Row():
                 cfg_normalization = gr.Checkbox(
                     label="CFG Normalization",
@@ -185,6 +205,21 @@ with gr.Blocks(css=css) as demo:
                     value=1.0
                 )
 
+            with gr.Row():
+                dynamic_shifting = gr.Checkbox(
+                    label="Use Dynamic Shifting",
+                    value=True
+                )
+                sigmas = gr.Dropdown(
+                    choices=[
+                        "Default",
+                        "Karras",
+                        "Exponential",
+                        "Beta"
+                    ],
+                    value="Default"
+                )
+        
         gr.Examples(examples=examples, inputs=[prompt], outputs=[result, seed], fn=infer, cache_examples=True, cache_mode="lazy")
 
     gr.on(
@@ -202,6 +237,9 @@ with gr.Blocks(css=css) as demo:
             system_prompt,
             cfg_normalization,
             cfg_trunc_ratio,
+            max_sequence_length,
+            dynamic_shifting,
+            sigmas
         ],
         outputs=[result, seed],
     )