mirror of
https://github.com/wassname/TTS.git
synced 2026-10-28 12:41:20 +08:00
Use pyworld for pitch
This commit is contained in:
1 parent
c1513ec4cd
commit
2c4bbbf9b9
6 files changed
+19
-163
No files matched your search
+4
-1
@@ -7,7 +7,10 @@ from TTS.utils.generic_utils import get_cuda
|
||||
def get_device_id():
|
||||
use_cuda, _ = get_cuda()
|
||||
if use_cuda:
|
||||
GPU_ID = "0"
|
||||
if 'CUDA_VISIBLE_DEVICES' in os.environ and os.environ['CUDA_VISIBLE_DEVICES'] != "":
|
||||
GPU_ID = os.environ['CUDA_VISIBLE_DEVICES'].split(',')[0]
|
||||
else:
|
||||
GPU_ID = "0"
|
||||
else:
|
||||
GPU_ID = ""
|
||||
return GPU_ID
|
||||
|
||||
Reference in new issue
Block a user