Spaces:
Running
on
T4
Running
on
T4
Update InferenceInterfaces/ControllableInterface.py
Browse files
InferenceInterfaces/ControllableInterface.py
CHANGED
@@ -17,10 +17,6 @@ class ControllableInterface:
|
|
17 |
else: # in this case we hopefully got a number.
|
18 |
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
|
19 |
os.environ["CUDA_VISIBLE_DEVICES"] = f"{gpu_id}"
|
20 |
-
if tts_model_path is None:
|
21 |
-
tts_model_path = hf_hub_download(repo_id="Flux9665/ToucanTTS", filename="ToucanTTS.pt")
|
22 |
-
if vocoder_model_path is None:
|
23 |
-
vocoder_model_path = hf_hub_download(repo_id="Flux9665/ToucanTTS", filename="Vocoder.pt")
|
24 |
if embedding_gan_path is None:
|
25 |
embedding_gan_path = hf_hub_download(repo_id="Flux9665/ToucanTTS", filename="embedding_gan.pt")
|
26 |
self.device = "cuda" if gpu_id != "cpu" else "cpu"
|
|
|
17 |
else: # in this case we hopefully got a number.
|
18 |
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
|
19 |
os.environ["CUDA_VISIBLE_DEVICES"] = f"{gpu_id}"
|
|
|
|
|
|
|
|
|
20 |
if embedding_gan_path is None:
|
21 |
embedding_gan_path = hf_hub_download(repo_id="Flux9665/ToucanTTS", filename="embedding_gan.pt")
|
22 |
self.device = "cuda" if gpu_id != "cpu" else "cpu"
|