Method forward

utils/unet_blocks.py:546–551

(self, input_tensor, temb, encoder_hidden_states, attention_mask=None, anchor_frame_idx=None)

            self.temporal_transformer.proj_out = zero_module(self.temporal_transformer.proj_out)

    def forward(self, input_tensor, temb, encoder_hidden_states, attention_mask=None, anchor_frame_idx=None):
        hidden_states = input_tensor
        hidden_states = self.temporal_transformer(hidden_states, encoder_hidden_states, attention_mask)

        output = hidden_states
        return output

class TemporalTransformer3DModel(nn.Module):
    def __init__(