(self, batch_size, max_seqlen, dtype=None, **kwargs)
| 460 | return x, residual |
| 461 | |
| 462 | def allocate_inference_cache(self, batch_size, max_seqlen, dtype=None, **kwargs): |
| 463 | return self.mixer.allocate_inference_cache( |
| 464 | batch_size, max_seqlen, dtype=dtype, **kwargs |
| 465 | ) |
| 466 | |
| 467 | |
| 468 | def create_block( |
nothing calls this directly
no outgoing calls
no test coverage detected