Update app.py

app.py

@@ -61,7 +61,7 @@ def download_file(url, folder_path, filename):
 # Download ESRGAN models
 download_file("https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealESRGAN_x2.pth?download=true", "models/upscalers/", "RealESRGAN_x2.pth")
 download_file("https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealESRGAN_x4.pth?download=true", "models/upscalers/", "RealESRGAN_x4.pth")
-
+lowstep_lora_repo = 'silveroxides/sdxl_lowstep_beta_v1'
 # Download the model files
 ckpt_dir_realpony = snapshot_download(repo_id="silveroxides/RNS_RealPonyV20")
 ckpt_dir_noobai = snapshot_download(repo_id="silveroxides/NoobAI-XL-EPS-1.0-Vwe")
@@ -207,12 +207,12 @@ def upscale_image(image, scale):
 
 @spaces.GPU(duration=120)
 def generate_image(model_choice, additional_positive_prompt, additional_negative_prompt, height, width, num_inference_steps,
-                   guidance_scale, pag_scale, pag_layers, num_images_per_prompt, use_random_seed, seed, sampler, clip_skip, 
+                   guidance_scale, pag_scale, pag_layers, num_images_per_prompt, use_lowstep_lora, use_random_seed, seed, sampler, clip_skip, 
                    use_florence2, use_medium_enhancer, use_long_enhancer,
                    use_positive_prefix, use_positive_suffix, use_negative_prefix, use_negative_suffix,
                    use_upscaler, upscale_factor,
                    input_image=None, progress=gr.Progress(track_tqdm=True)):
-    
+
     # Select the appropriate pipe based on the model choice
     if model_choice == "Real Pony RNS":
         pipe = pipe_realpony
@@ -220,12 +220,15 @@ def generate_image(model_choice, additional_positive_prompt, additional_negative
         pipe = pipe_noobai
     else:  # "Hybrid Pony SDXL"
         pipe = pipe_hybridpony
-    
+
+    if use_lowstep_lora:
+        pipe.load_lora_weights(lowstep_lora_repo, weight_name='pytorch_lora_weights.safetensors')
+
     if use_random_seed:
         seed = random.randint(0, 2**32 - 1)
     else:
         seed = int(seed)  # Ensure seed is an integer
-    
+
     # Set the scheduler based on the selected sampler
     pipe.scheduler = samplers[sampler]
     
@@ -331,6 +334,7 @@ with gr.Blocks(theme='bethecloud/storj_theme') as demo:
                 pag_scale = gr.Number(label="Pag Scale", value=3.0)
                 pag_layers = gr.Dropdown(label="Model layers to apply Pag to", info="mid is the one used on the paper, up and down blocks seem unstable", choices=["up", "mid", "down"], multiselect=True, value="mid")
                 num_images_per_prompt = gr.Slider(1, 4, 1, step=1, label="Number of images per prompt")
+                use_lowstep_lora = gr.Checkbox(label="Use Lowstep LoRA", value=False)
                 use_random_seed = gr.Checkbox(label="Use Random Seed", value=True)
                 seed = gr.Number(label="Seed", value=0, precision=0)
                 sampler = gr.Dropdown(label="Sampler", choices=list(samplers.keys()), value="Euler a")
@@ -381,7 +385,7 @@ with gr.Blocks(theme='bethecloud/storj_theme') as demo:
         inputs=[
             model_choice,  # Add this new input
             positive_prompt, negative_prompt, height, width, num_inference_steps,
-            guidance_scale, pag_scale, pag_layers, num_images_per_prompt, use_random_seed, seed, sampler,
+            guidance_scale, pag_scale, pag_layers, num_images_per_prompt, use_lowstep_lora, use_random_seed, seed, sampler,
             clip_skip, use_florence2, use_medium_enhancer, use_long_enhancer,
             use_positive_prefix, use_positive_suffix, use_negative_prefix, use_negative_suffix,
             use_upscaler, upscale_factor,