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