up

app.py

@@ -26,7 +26,7 @@ def predict(dict, prompt="", guidance_scale=7.5, steps=20, strength=1.0, schedul
         add_kwargs["algorithm_type"] = "sde-dpmsolver++"
 
     scheduler = getattr(diffusers, scheduler_class_name)
-    pipe.scheduler = scheduler.from_config(pipe.scheduler.config, **add_kwargs)
+    pipe.scheduler = scheduler.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", subfolder="scheduler", **add_kwargs)
     
     init_image = dict["image"].convert("RGB").resize((1024, 1024))
     mask = dict["mask"].convert("RGB").resize((1024, 1024))
@@ -90,6 +90,8 @@ with image_blocks as demo:
                         guidance_scale = gr.Number(value=7.5, minimum=1.0, maximum=20.0, step=0.1, label="guidance_scale")
                         steps = gr.Number(value=20, minimum=10, maximum=50, step=0.1, label="steps")
                         strength = gr.Number(value=1.0, minimum=0.0, maximum=1.0, step=0.05, label="strength")
+
+                    with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
                         schedulers = ["DEISMultistepScheduler", "HeunDiscreteScheduler", "EulerDiscreteScheduler", "DPMSolverMultistepScheduler", "DPMSolverMultistepScheduler-Karras", "DPMSolverMultistepScheduler-Karras-SDE"]
                         scheduler = gr.Dropdown(choices=schedulers, value="EulerDiscreteScheduler")
                         
@@ -118,4 +120,4 @@ with image_blocks as demo:
                 """
             )
 
-image_blocks.launch(share=True)
+image_blocks.launch()