(self, inputs)
| 263 | self.model = model |
| 264 | |
| 265 | def forward(self, inputs): |
| 266 | x_chunk, timesteps, context, attn_mask = inputs |
| 267 | out = self.model(x_chunk, timesteps, context=context, attention_mask=attn_mask) |
| 268 | return out |
| 269 | |
| 270 | |
| 271 | # class NativeWrapper(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected