jit level

mindspore-lab · Sep 25, 2024 · d9eed78 · d9eed78
1 parent 6c98961
commit d9eed78
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)