Update app.py
Browse files
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)
|