blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-06 03:57:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

mps support

Browse Source
This commit is contained in:
Wu Zichen 2024-01-24 17:30:49 +08:00
parent ed05443cfe
commit a4c22b24f8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
GPT_SoVITS/s1_train.py
View File

@ -116,9 +116,9 @@ def main(args):
devices=-1,
benchmark=False,
fast_dev_run=False,
strategy=DDPStrategy(
strategy = "auto" if torch.mps.is_available() else DDPStrategy(
process_group_backend="nccl" if platform.system() != "Windows" else "gloo"
),
), # mps 不支持多节点训练
precision=config["train"]["precision"],
logger=logger,
num_sanity_val_steps=0,