Lengyue 2 лет назад
Родитель
Сommit
cbab5b8ec8
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      fish_speech/models/text2semantic/lit_module.py

+ 1 - 1
fish_speech/models/text2semantic/lit_module.py

@@ -189,7 +189,7 @@ class TextToSemantic(L.LightningModule):
             # We want to shift the labels by one to the right
             codebook_labels = labels[:, 1 : 1 + self.model.config.num_codebooks, :-1]
             codebook_labels = torch.nn.functional.pad(
-                codebook_labels, (0, 1), value=-100
+                codebook_labels, (1, 0), value=-100
             ).mT
 
             semantic_loss = F.cross_entropy(