(
self,
input_tensor,
temb,
encoder_hidden_states,
attention_mask=None,
anchor_frame_idx=None,
)
| 74 | ) |
| 75 | |
| 76 | def forward( |
| 77 | self, |
| 78 | input_tensor, |
| 79 | temb, |
| 80 | encoder_hidden_states, |
| 81 | attention_mask=None, |
| 82 | anchor_frame_idx=None, |
| 83 | ): |
| 84 | hidden_states = input_tensor |
| 85 | hidden_states = self.temporal_transformer( |
| 86 | hidden_states, encoder_hidden_states, attention_mask |
| 87 | ) |
| 88 | |
| 89 | output = hidden_states |
| 90 | return output |
| 91 | |
| 92 | |
| 93 | class TemporalTransformer3DModel(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected