Update app.py

app.py

@@ -507,26 +507,27 @@ def generate_video(
             except Exception as e:
                 print(f"[LoRA] pipeline rebuild FAILED: {e}")
 
-        # Force transformer rebuild so LoRA strength actually applies
+        # Reset transformer so next call rebuilds it with new LoRA strength (NO preloading!)
         try:
             if hasattr(pipeline, "model_ledger"):
-                # Hard reset ALL known transformer references
                 if hasattr(pipeline.model_ledger, "_transformer"):
+                    del pipeline.model_ledger._transformer
                     pipeline.model_ledger._transformer = None
         
-                # VERY IMPORTANT: also clear pipeline components cache if present
-                if hasattr(pipeline, "pipeline_components"):
-                    try:
-                        pipeline.pipeline_components = None
-                    except Exception:
-                        pass
+            if hasattr(pipeline, "pipeline_components"):
+                try:
+                    del pipeline.pipeline_components
+                    pipeline.pipeline_components = None
+                except Exception:
+                    pass
         
-                # Force rebuild NOW (so it uses the new LoRA strength)
-                _ = pipeline.model_ledger.transformer()
+            # CRITICAL: force cleanup BEFORE rebuild happens
+            cleanup_memory()
+            torch.cuda.empty_cache()
         
-                print("[LoRA] transformer force-rebuilt with new strength")
+            print("[LoRA] transformer reset; will rebuild during inference")
         except Exception as e:
-            print(f"[LoRA] transformer rebuild failed: {e}")
+            print(f"[LoRA] transformer reset failed: {e}")
         
         log_memory("before pipeline call")