Flux9665 commited on
Commit
d5f2d74
·
verified ·
1 Parent(s): cf8ad0c

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"