This commit is contained in:
lyuxiang.lx
2024-10-16 13:30:13 +08:00
parent 29507bc77a
commit 7e6d60c24c
2 changed files with 4 additions and 4 deletions

View File

@@ -99,7 +99,7 @@ def main():
'prompt_speech_feat': speech_feat, 'prompt_speech_feat_len': speech_feat_len,
'llm_embedding': utt_embedding, 'flow_embedding': utt_embedding}
tts_speeches = []
for model_output in model.inference(**model_input):
for model_output in model.tts(**model_input):
tts_speeches.append(model_output['tts_speech'])
tts_speeches = torch.concat(tts_speeches, dim=1)
tts_key = '{}_{}'.format(utts[0], tts_index[0])