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)