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:
Kakaru 2025-10-20 20:44:34 +08:00 committed by GitHub
parent 70a9243285
commit 917f73c38c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
GPT_SoVITS/s1_train.py
View File

@ -29,6 +29,7 @@ import musa_utils
if musa_utils.is_available(): if musa_utils.is_available():
import musa_accelerator import musa_accelerator
os.environ["MUSA_VISIBLE_DEVICES"] = os.environ["_MUSA_VISIBLE_DEVICES"] os.environ["MUSA_VISIBLE_DEVICES"] = os.environ["_MUSA_VISIBLE_DEVICES"]
os.environ["MUSA_LAUNCH_BLOCKING"] = "1"
class my_model_ckpt(ModelCheckpoint): class my_model_ckpt(ModelCheckpoint):
@ -183,4 +184,4 @@ if __name__ == "__main__":
args = parser.parse_args() args = parser.parse_args()
logging.info(str(args)) logging.info(str(args))
main(args) main(args)