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

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

View File

@ -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)
main(args)