Update app.py

make hyper flux lora optional

app.py

@@ -20,7 +20,8 @@ pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev",
                                          custom_pipeline="pipeline_flux_rf_inversion",
                                           torch_dtype=torch.bfloat16)
 pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), lora_scale=0.125)
-pipe.fuse_lora(lora_scale=0.125)
+#pipe.fuse_lora(lora_scale=0.125)
+pipe.enable_lora()
 pipe.to(DEVICE)
 
 examples = [[Image.open("cat.jpg"), "a tiger", 0, 4, 0.5, 0.9, False, 1, 8, 8, 789385745, False],
@@ -38,6 +39,14 @@ def resize_img(image, max_size=1024):
 def check_style(stylezation):
     return True, 2
 
+def check_hyper_flux_lora(enable_hyper_flux):
+    if enable_hyper_flux:
+        pipe.enable_lora()
+        return 8, 8, 4
+    else:
+        pipe.disable_lora()
+        return 28, 28, 7
+
 @spaces.GPU(duration=85)
 def invert_and_edit(image, 
                     prompt, 
@@ -117,7 +126,9 @@ based on the implementations of [@raven38](https://github.com/raven38) & [@DarkM
                     max_lines=1,
                     placeholder="describe the edited output",
                 )
-                stylezation = gr.Checkbox(label="stylzation")
+                with gr.Row():
+                    enable_hyper_flux = gr.Checkbox(label="8-step LoRA", value=True)
+                    stylezation = gr.Checkbox(label="stylzation")
                 with gr.Row():          
                     start_timestep = gr.Slider(
                     label="start timestep",
@@ -265,6 +276,12 @@ based on the implementations of [@raven38](https://github.com/raven38) & [@DarkM
         inputs=[stylezation],
         outputs=[eta_decay, decay_power]
     )
+
+    enable_hyper_flux.change(
+        fn=check_hyper_flux_lora,
+        inputs=[enable_hyper_flux],
+        outputs=[num_inversion_steps, num_inference_steps, stop_timestep]
+    )