xlgeng commited on
Commit
cc344d5
·
1 Parent(s): 841f290

开始部署

Browse files
Files changed (1) hide show
  1. tts/cosyvoice/cli/model.py +1 -1
tts/cosyvoice/cli/model.py CHANGED
@@ -66,7 +66,7 @@ class CosyVoiceModel:
66
  # rtf and decoding related
67
  self.stream_scale_factor = 1
68
  assert self.stream_scale_factor >= 1, 'stream_scale_factor should be greater than 1, change it according to your actual rtf'
69
- self.llm_context = torch.cuda.stream(torch.cuda.Stream(self.device)) if torch.cuda.is_available() else nullcontext()
70
  self.lock = threading.Lock()
71
  # dict used to store session related variable
72
  self.tts_speech_token_dict = {}
 
66
  # rtf and decoding related
67
  self.stream_scale_factor = 1
68
  assert self.stream_scale_factor >= 1, 'stream_scale_factor should be greater than 1, change it according to your actual rtf'
69
+ self.llm_context = nullcontext()
70
  self.lock = threading.Lock()
71
  # dict used to store session related variable
72
  self.tts_speech_token_dict = {}