deepgeek/CosyVoice
2
0
Fork 0
mirror of https://github.com/FunAudioLLM/CosyVoice.git synced 2026-02-04 09:29:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "fix triton token2wav model cache thread unsafety"

Browse Source 
This reverts commit cd26dd1932.
This commit is contained in:
김의진
2025-10-27 18:07:30 +09:00
parent cd26dd1932
commit fa2781405f
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
runtime/triton_trtllm/model_repo/token2wav/1/model.py
View File

@@ -28,7 +28,6 @@ import json
import os import os
import logging import logging
from uuid import uuid4
import torch import torch
from torch.utils.dlpack import to_dlpack from torch.utils.dlpack import to_dlpack
@@ -236,17 +235,17 @@ class TritonPythonModel:
stream = True stream = True
else: else:
stream = False stream = False
uuid = uuid4().hex request_id = request.request_id()
audio_hat = self.token2wav_model.model.token2wav(token=target_speech_tokens, audio_hat = self.token2wav_model.model.token2wav(token=target_speech_tokens,
prompt_token=prompt_speech_tokens, prompt_token=prompt_speech_tokens,
prompt_feat=prompt_speech_feat, prompt_feat=prompt_speech_feat,
embedding=prompt_spk_embedding, embedding=prompt_spk_embedding,
token_offset=token_offset, token_offset=token_offset,
uuid=uuid, uuid=request_id,
stream=stream, stream=stream,
finalize=finalize) finalize=finalize)
if finalize: if finalize:
self.token2wav_model.model.hift_cache_dict.pop(uuid) self.token2wav_model.model.hift_cache_dict.pop(request_id)
else: else:
tts_mel, _ = self.token2wav_model.model.flow.inference( tts_mel, _ = self.token2wav_model.model.flow.inference(