Update app.py

app.py

@@ -95,6 +95,7 @@ def run(
     num_steps=25,
     guidance_scale=5,
     adapter_conditioning_scale=0.8,
+    cond_tau=0.8,
     seed=0,
 ) -> PIL.Image.Image:
     image = image.convert("RGB").resize((1024, 1024))
@@ -112,6 +113,7 @@ def run(
         generator=generator,
         guidance_scale=guidance_scale,
         adapter_conditioning_scale=adapter_conditioning_scale,
+        cond_tau=cond_tau,
     ).images[0]
     return out
 
@@ -126,10 +128,9 @@ with gr.Blocks() as demo:
                 type="pil",
                 image_mode="1",
                 invert_colors=True,
-                # shape=(1024, 1024),
+                shape=(1024, 1024),
                 brush_radius=4,
-                height=1024,
-                width=1024,
+                height=600,
             )
             prompt = gr.Textbox(label="Prompt")
             run_button = gr.Button("Run")
@@ -157,11 +158,18 @@ with gr.Blocks() as demo:
                     value=5,
                 )
                 adapter_conditioning_scale = gr.Slider(
-                    label="Adapter Conditioning Ccale",
+                    label="Adapter Conditioning Scale",
                     minimum=0.5,
                     maximum=1,
                     step=0.1,
-                    value=.85,
+                    value=.8,
+                )
+                cond_tau = gr.Slider(
+                    label="Fraction of timesteps for which adapter should be applied",
+                    minimum=0.5,
+                    maximum=1,
+                    step=0.1,
+                    value=.8,
                 )
                 seed = gr.Slider(
                     label="Seed",
@@ -182,6 +190,7 @@ with gr.Blocks() as demo:
         num_steps,
         guidance_scale,
         adapter_conditioning_scale,
+        cond_tau,
         seed,
     ]
     prompt.submit(