Update train.py

This commit is contained in:
jinzr 2024-03-01 09:40:43 +08:00
parent a0dc097ad9
commit 8f8e516e29

View File

@ -342,7 +342,7 @@ def prepare_input(
torch.Tensor([speaker_map[sid] for sid in batch["speakers"]]).int().to(device) torch.Tensor([speaker_map[sid] for sid in batch["speakers"]]).int().to(device)
) )
tokens = tokenizer.texts_to_token_ids( tokens = tokenizer.tokens_to_token_ids(
tokens, intersperse_blank=True, add_sos=True, add_eos=True tokens, intersperse_blank=True, add_sos=True, add_eos=True
) )
tokens = k2.RaggedTensor(tokens) tokens = k2.RaggedTensor(tokens)