dongyh20 commited on
Commit
c2fdddf
·
1 Parent(s): 2658df1

update space

Browse files
ola/CosyVoice_main/cosyvoice/cli/model.py CHANGED
@@ -57,7 +57,7 @@ class CosyVoiceModel:
57
  # rtf and decoding related
58
  self.stream_scale_factor = 1
59
  assert self.stream_scale_factor >= 1, 'stream_scale_factor should be greater than 1, change it according to your actual rtf'
60
- self.llm_context = torch.cuda.stream(torch.cuda.Stream(self.device)) if torch.cuda.is_available() else nullcontext()
61
  self.lock = threading.Lock()
62
  # dict used to store session related variable
63
  self.tts_speech_token_dict = {}
 
57
  # rtf and decoding related
58
  self.stream_scale_factor = 1
59
  assert self.stream_scale_factor >= 1, 'stream_scale_factor should be greater than 1, change it according to your actual rtf'
60
+ self.llm_context = torch.cuda.stream(torch.cuda.Stream(self.device))
61
  self.lock = threading.Lock()
62
  # dict used to store session related variable
63
  self.tts_speech_token_dict = {}