diff --git a/cosyvoice/flow/flow_matching.py b/cosyvoice/flow/flow_matching.py index d3beb9ec2..3f992cdbd 100644 --- a/cosyvoice/flow/flow_matching.py +++ b/cosyvoice/flow/flow_matching.py @@ -170,7 +170,7 @@ def compute_loss(self, x1, mask, mu, spks=None, cond=None, streaming=False): y: conditional flow shape: (batch_size, n_feats, mel_timesteps) """ - b, _, t = mu.shape + b, _, _ = mu.shape # random timestep t = torch.rand([b, 1, 1], device=mu.device, dtype=mu.dtype)