Update app.py

app.py

@@ -63,6 +63,8 @@ pipe = DiffusionPipeline.from_pretrained(
 LIGHTNING_LORA_REPO = "lightx2v/Qwen-Image-Lightning"
 LIGHTNING_LORA_WEIGHT = "Qwen-Image-Lightning-4steps-V2.0-bf16.safetensors"
 LIGHTNING8_LORA_WEIGHT = "Qwen-Image-Lightning-8steps-V2.0-bf16.safetensors"
+LIGHTNING_FP8_4STEPS_LORA_WEIGHT = "Qwen-Image-fp8-e4m3fn-Lightning-4steps-V1.0-bf16.safetensors"
+
 
 MAX_SEED = np.iinfo(np.int32).max
 
@@ -234,6 +236,26 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, asp
                 adapter_name="style"
             )
             
+            # Set both adapters active with their weights
+            pipe.set_adapters(["lightning", "style"], adapter_weights=[1.0, lora_scale])
+    elif speed_mode == "light 4 fp8":
+        with calculateDuration("Loading Lightning LoRA and style LoRA"):
+            # Load Lightning LoRA first
+            pipe.load_lora_weights(
+                LIGHTNING_LORA_REPO, 
+                weight_name=LIGHTNING_FP8_4STEPS_LORA_WEIGHT,
+                adapter_name="lightning"
+            )
+            
+            # Load the selected style LoRA
+            weight_name = selected_lora.get("weights", None)
+            pipe.load_lora_weights(
+                lora_path, 
+                weight_name=weight_name, 
+                low_cpu_mem_usage=True,
+                adapter_name="style"
+            )
+            
             # Set both adapters active with their weights
             pipe.set_adapters(["lightning", "style"], adapter_weights=[1.0, lora_scale])
     elif speed_mode == "light 8":
@@ -300,12 +322,12 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, asp
     # Formatear para Gallery: (img, "Seed: N")
     #images_for_gallery = [(img, f"Seed: {s}") for (img, s) in pairs]
     images_for_gallery = [
-    (
-        img,
-        f"<div style='cursor:pointer;color:#ccc' onclick='navigator.clipboard.writeText(\"{s}\");alert(\"Seed {s} copied!\")'>Seed: {s}</div>"
-    )
-    for (img, s) in pairs
-]
+        (
+            img,
+            s
+        )
+        for (img, s) in pairs
+    ]
 
 
     # Debe devolver DOS valores porque outputs=[result, seed]
@@ -519,7 +541,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, delete_cache=(60, 60)) as app:
                 with gr.Column():
                     speed_mode = gr.Radio(
                         label="Generation Mode",
-                        choices=["light 4", "light 8", "normal"],
+                        choices=["light 4", "light 4 fp8", "light 8", "normal"],
                         value="light 4",
                         info="'light' modes use Lightning LoRA for faster generation"
                     )