Update app.py

app.py

@@ -20,8 +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.enable_lora()
+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],
@@ -41,10 +41,10 @@ def check_style(stylezation):
 
 def check_hyper_flux_lora(enable_hyper_flux):
     if enable_hyper_flux:
-        pipe.enable_lora()
+        pipe.fuse_lora(lora_scale=0.125)
         return 8, 8, 4
     else:
-        pipe.disable_lora()
+        pipe.unfuse_lora()
         return 28, 28, 7
 
 @spaces.GPU(duration=85)