Update t2s_model.py (#1283)

y is None at no reference mode, and Nonetype object has no attribute shape
2024-07-08 20:51:15 +08:00
parent 8d0cf0f8a9
commit 923af471d2
1 changed files with 2 additions and 1 deletions
--- a/GPT_SoVITS/AR/models/t2s_model.py
+++ b/GPT_SoVITS/AR/models/t2s_model.py
@@ -528,6 +528,7 @@ class Text2SemanticDecoder(nn.Module):
            y_pos = None
            xy_pos = x
            y = torch.zeros(x.shape[0], 0, dtype=torch.int, device=x.device)
+            prompts = y
            ref_free = True

        x_attn_mask_pad = F.pad(