Method forward

diffsynth/models/stepvideo_text_encoder.py:508–520

(
        self,
        input_ids=None,
        attention_mask=None,
    )

        self.transformer = Transformer(config)

    def forward(
        self,
        input_ids=None,
        attention_mask=None,
    ):

        hidden_states = self.tok_embeddings(input_ids)

        hidden_states = self.transformer(
            hidden_states,
            attention_mask,
        )
        return hidden_states