Update app.py

app.py

@@ -56,16 +56,16 @@ def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, model=D
         vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
         controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float16)
         repo = StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
-        steps = steps or 16
-        guidance = guidance or 7
+        steps = (not steps or steps < 0 and 16) or steps
+        guidance = (not guidance or guidance < 0 and 7) or guidance
     else:
         vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
         controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float16)
         repo = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
         repo.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
         repo.set_adapters(["base"], adapter_weights=[0.7])
-        steps = steps or 16
-        guidance = guidance or 3
+        steps = (not steps or steps < 0 and 16) or steps
+        guidance = (not guidance or guidance < 0 and 3) or guidance
         
     repo.to(DEVICE)
     
@@ -104,8 +104,8 @@ with gr.Blocks(css=css) as main:
         model = gr.Dropdown(label="Models", choices=["Default", "Anime"], value="Default")
         height = gr.Slider(minimum=1, maximum=2160, step=1, value=DEFAULT_HEIGHT, label="Height")
         width = gr.Slider(minimum=1, maximum=2160, step=1, value=DEFAULT_WIDTH, label="Width")
-        steps = gr.Slider(minimum=0, maximum=100, step=1, value=16, label="Steps")
-        guidance = gr.Slider(minimum=0, maximum=100, step=0.001, value=3, label = "Guidance")
+        steps = gr.Slider(minimum=-1, maximum=100, step=1, value=16, label="Steps")
+        guidance = gr.Slider(minimum=-1, maximum=100, step=0.001, value=3, label = "Guidance")
         number = gr.Slider(minimum=1, maximum=4, step=1, value=1, label="Number")
         seed = gr.Textbox(lines=1, value="", label="Seed (Blank for random)")
         submit = gr.Button("▶")