(self, x_or_x_list)
| 286 | return attn_bias.split(x) |
| 287 | |
| 288 | def forward(self, x_or_x_list): |
| 289 | if isinstance(x_or_x_list, Tensor): |
| 290 | return super().forward(x_or_x_list) |
| 291 | elif isinstance(x_or_x_list, list): |
| 292 | if not XFORMERS_AVAILABLE: |
| 293 | raise AssertionError("xFormers is required for using nested tensors") |
| 294 | return self.forward_nested(x_or_x_list) |
| 295 | else: |
| 296 | raise AssertionError |
nothing calls this directly
no test coverage detected