gz412 commited on
Commit
1dfc874
·
1 Parent(s): bd84a81

test app.py

Browse files
Files changed (1) hide show
  1. cosyvoice/cli/model.py +1 -1
cosyvoice/cli/model.py CHANGED
@@ -260,7 +260,7 @@ class CosyVoice2Model(CosyVoiceModel):
260
  # speech fade in out
261
  self.speech_window = np.hamming(2 * self.source_cache_len)
262
  # rtf and decoding related
263
- self.llm_context = torch.cuda.stream(torch.cuda.Stream(self.device)) if torch.cuda.is_available() else nullcontext()
264
  self.lock = threading.Lock()
265
  # dict used to store session related variable
266
  self.tts_speech_token_dict = {}
 
260
  # speech fade in out
261
  self.speech_window = np.hamming(2 * self.source_cache_len)
262
  # rtf and decoding related
263
+ self.llm_context = nullcontext()
264
  self.lock = threading.Lock()
265
  # dict used to store session related variable
266
  self.tts_speech_token_dict = {}