ceymox commited on
Commit
6e47a0b
·
verified ·
1 Parent(s): c814750

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ def load_tts_model_with_retry(max_retries=3, retry_delay=5):
88
  tts_repo_id,
89
  trust_remote_code=True,
90
  local_files_only=True,
91
- device_map="auto"
92
  torch_dtype=torch.float16
93
  )
94
  tts_model_wrapper = TTSModelWrapper(tts_model)
 
88
  tts_repo_id,
89
  trust_remote_code=True,
90
  local_files_only=True,
91
+ device_map="auto",
92
  torch_dtype=torch.float16
93
  )
94
  tts_model_wrapper = TTSModelWrapper(tts_model)