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"
|