Method encode

opensora/models/text_encoder/t5.py:194–197

(self, input_ids, attention_mask=None)

        return self.t5.tokenize_fn

    def encode(self, input_ids, attention_mask=None):
        caption_embs, emb_masks = self.t5.get_text_embeddings(input_ids, attention_mask)
        caption_embs = caption_embs[:, None]
        return dict(y=caption_embs, mask=emb_masks)

    def null(self, n):
        null_y = self.y_embedder.y_embedding[None].repeat(n, 1, 1)[:, None]