Update app.py

app.py

@@ -177,6 +177,8 @@ except Exception as e:
     print(f"[PATCH 5] Could not apply key mapping patch: {e}")
 
 # --- PATCH 7: MATERIALIZE LEFTOVER META TENSORS (unused vision_tower / rotary buffers) ---
+# Also manually restores the cached_property caching contract that ModelLedger.text_encoder
+# originally had, since replacing the class attribute with a plain function loses it.
 try:
     from ltx_pipelines.utils import model_ledger as _ml
 
@@ -203,18 +205,26 @@ try:
                 new_buf = torch.zeros(b.shape, dtype=b.dtype, device=device)
                 parent.register_buffer(leaf, new_buf, persistent=False)
 
-
     def _patched_text_encoder(self):
+        # Manually replicate cached_property behavior: reuse the cached instance
+        # from __dict__ if it's already been built, exactly like the original
+        # ModelLedger contract documents (cleared only via clear_vram()).
+        if "text_encoder" in self.__dict__:
+            return self.__dict__["text_encoder"]
+
         model = self.text_encoder_builder.build(device=self._target_device(), dtype=self.dtype)
         target_device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         _materialize_meta_(model, target_device)   # fix only the meta leftovers
-        return model.to(target_device).eval()
+        model = model.to(target_device).eval()
+
+        self.__dict__["text_encoder"] = model  # cache it, same as the original cached_property
+        return model
 
     _ml.ModelLedger.text_encoder = _patched_text_encoder
     print("[PATCH 7] Patched ModelLedger.text_encoder to materialize leftover meta tensors "
-          "(unused Gemma3 vision_tower / rotary_emb buffers) instead of failing on .to()")
+          "AND restore caching (instance __dict__) so it isn't rebuilt on every call.")
 except Exception as e:
-    print(f"[PATCH 7] Could not patch ModelLedger.text_encoder: {e}")
+    print(f"[PATCH 7] Could not apply ModelLedger.text_encoder patch: {e}")
     
 
 logging.getLogger().setLevel(logging.INFO)