Update LTX-Video/ltx_video/pipelines/pipeline_ltx_video.py

LTX-Video/ltx_video/pipelines/pipeline_ltx_video.py

@@ -426,12 +426,12 @@ class LTXVideoPipeline(DiffusionPipeline):
 
         self.allowed_inference_steps = allowed_inference_steps
         
-        self.spy = SpyLatent(vae=self.pipeline.vae)
+        self.spy = SpyLatent(vae=vae)
 
     def mask_text_embeddings(self, emb, mask):
         if emb.shape[0] == 1:
             keep_index = mask.sum().item()
-            return emb[:, :, :keep_index, :], keep_index
+            return emb[, :, :keep_index, :], keep_index
         else:
             masked_feature = emb * mask[:, None, :, None]
             return masked_feature, emb.shape[2]