Update app.py

app.py

@@ -343,6 +343,7 @@ class SDXLFlowMatchingPipeline:
                 print("[Lyra Debug] Normalizing CLIP-G reconstruction to match input stats")
                 lyra_clip_g = (lyra_clip_g - lyra_clip_g.mean()) / (lyra_clip_g.std() + 1e-8)
                 lyra_clip_g = lyra_clip_g * clip_g_embeds.std() + clip_g_embeds.mean()
+
         
         # Blend original CLIP with Lyra reconstruction
         fused_clip_l = (1 - lyra_strength) * clip_l_embeds + lyra_strength * lyra_clip_l
@@ -356,7 +357,7 @@ class SDXLFlowMatchingPipeline:
         # Process negative prompt (simpler - just use original CLIP for negative)
         if negative_prompt:
             # For negative, blend less aggressively
-            neg_strength = lyra_strength * 0.5
+            neg_strength = lyra_strength
             
             t5_neg_prompt = f"{negative_prompt} {SUMMARY_SEPARATOR} {negative_prompt}"
             t5_inputs_neg = self.t5_tokenizer(
@@ -432,7 +433,7 @@ class SDXLFlowMatchingPipeline:
         use_lyra: bool = False,
         clip_skip: int = 1,
         t5_summary: str = "",
-        lyra_strength: float = 0.3,
+        lyra_strength: float = 1.0,
         progress_callback=None
     ):
         """Generate image using SDXL architecture."""
@@ -1053,7 +1054,7 @@ def load_lyra_vae_xl(
         print("  📥 Downloading model.pt...")
         checkpoint_path = hf_hub_download(
             repo_id=repo_id,
-            filename="model.pt",
+            filename="checkpoint_lyra_illustrious_37000.pt",
             repo_type="model"
         )
         
@@ -1450,10 +1451,10 @@ def create_demo():
                 lyra_strength = gr.Slider(
                     label="Lyra Blend Strength",
                     minimum=0.0,
-                    maximum=1.0,
-                    value=0.3,
+                    maximum=3.0,
+                    value=1.0,
                     step=0.05,
-                    info="0.0 = pure CLIP, 1.0 = pure Lyra reconstruction"
+                    info="0.0 = pure CLIP, 1.0 = pure Lyra reconstruction, 3.0 = way too much but try it anyway"
                 )
                 
                 with gr.Accordion("Generation Settings", open=True):
@@ -1556,21 +1557,21 @@ def create_demo():
                     "A beautiful anime girl with flowing blue hair wearing a school uniform, surrounded by delicate pink cherry blossoms against a bright sky",
                     "lowres, bad anatomy, worst quality, low quality",
                     "Illustrious XL",
-                    2, 25, 7.0, 1024, 1024, 0.0, False, True, 0.3, 42, False
+                    2, 25, 7.0, 1024, 1024, 0.0, False, True, 0.8, 42, False
                 ],
                 [
                     "A majestic mountain landscape at golden hour, crystal clear lake, photorealistic, 8k",
                     "A breathtaking mountain vista bathed in warm golden light at sunset, with a perfectly still crystal clear lake reflecting the peaks",
                     "blurry, low quality",
                     "SDXL Base",
-                    1, 30, 7.5, 1024, 1024, 0.0, False, True, 0.3, 123, False
+                    1, 30, 7.5, 1024, 1024, 0.0, False, True, 0.8, 123, False
                 ],
                 [
                     "cyberpunk city at night, neon lights, rain, highly detailed",
                     "A futuristic cyberpunk metropolis at night with vibrant neon lights reflecting off rain-slicked streets",
                     "low quality, blurry",
                     "Flow-Lune (SD1.5)",
-                    1, 20, 7.5, 512, 512, 2.5, True, True, 0.3, 456, False
+                    1, 20, 7.5, 512, 512, 2.5, True, True, 0.8, 456, False
                 ],
             ],
             inputs=[