From 917f73c38c2e9a05cb963e0b5be55154cfd63801 Mon Sep 17 00:00:00 2001
From: Kakaru <97896816+KakaruHayate@users.noreply.github.com>
Date: Mon, 20 Oct 2025 20:44:34 +0800
Subject: [PATCH] may fix RuntimeError: MUSA error: an illegal memory access
 was encountered

---
 GPT_SoVITS/s1_train.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/GPT_SoVITS/s1_train.py b/GPT_SoVITS/s1_train.py
index ba8b4cbb..32696379 100644
--- a/GPT_SoVITS/s1_train.py
+++ b/GPT_SoVITS/s1_train.py
@@ -29,6 +29,7 @@ import musa_utils
 if musa_utils.is_available():
     import musa_accelerator
     os.environ["MUSA_VISIBLE_DEVICES"] = os.environ["_MUSA_VISIBLE_DEVICES"]
+    os.environ["MUSA_LAUNCH_BLOCKING"] = "1"
 
 
 class my_model_ckpt(ModelCheckpoint):
@@ -183,4 +184,4 @@ if __name__ == "__main__":
 
     args = parser.parse_args()
     logging.info(str(args))
-    main(args)
\ No newline at end of file
+    main(args)