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

LTX-Video/ltx_video/pipelines/pipeline_ltx_video.py

@@ -431,7 +431,7 @@ class LTXVideoPipeline(DiffusionPipeline):
     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]