(self, data)
| 303 | |
| 304 | @T.no_grad() |
| 305 | def forward(self, data): |
| 306 | if self.c.split: |
| 307 | x1, x2 = data.chunk(2, dim=-1) |
| 308 | x = self.input(x1) + self.input2(x2) |
| 309 | else: |
| 310 | x = self.input(data) |
| 311 | |
| 312 | for l, layer in enumerate(self.layers): |
| 313 | x = layer(x, kv=self.cache[l]) |
| 314 | |
| 315 | if self.c.split: |
| 316 | return self.output(x), self.output2(x) |
| 317 | else: |
| 318 | return self.output(x) |
| 319 | |
| 320 | @T.no_grad() |
| 321 | def next_audio_from_audio(self, audio_data: T.Tensor, temps=(0.8, (0.5, 0.1))): |
no test coverage detected