blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-12-16 17:40:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

may fix RuntimeError: MUSA error: an illegal memory access was encountered

Browse Source
This commit is contained in:
KakaruHayate 2025-10-19 16:49:13 +08:00
parent 90141d2029
commit 06994f2a13
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
GPT_SoVITS/s2_train.py
View File

@ -17,6 +17,7 @@ import torch.multiprocessing as mp
from torch.cuda.amp import GradScaler from torch.cuda.amp import GradScaler
musa_ddp = False musa_ddp = False
if musa_utils.is_available(): if musa_utils.is_available():
os.environ["MUSA_LAUNCH_BLOCKING"] = 1
autocast = torch.musa.amp.autocast autocast = torch.musa.amp.autocast
musa_ddp = musa_utils.should_ddp() musa_ddp = musa_utils.should_ddp()
elif torch.cuda.is_available(): elif torch.cuda.is_available():