diff --git a/examples/kohya_sd_scripts/sdxl_train_network.py b/examples/kohya_sd_scripts/sdxl_train_network.py
index 16fcaf2f53..1c7f4b6a90 100644
--- a/examples/kohya_sd_scripts/sdxl_train_network.py
+++ b/examples/kohya_sd_scripts/sdxl_train_network.py
@@ -149,7 +149,7 @@ def setup_parser() -> argparse.ArgumentParser:
     args = parser.parse_args()
     train_util.verify_command_line_training_args(args)
     args = train_util.read_config_from_file(args, parser)
-    ms.set_context(mode=ms.GRAPH_MODE, jit_syntax_level=ms.STRICT, jit_config={"jit_level": "O2"})
+    ms.set_context(mode=ms.GRAPH_MODE, jit_syntax_level=ms.STRICT, jit_config={"jit_level": "O1"})
     train_util.init_distributed_device(args)
     trainer = SdxlNetworkTrainer()
     trainer.train(args)