adapted to ms2.6

Author: QiuYuFong

examples/dit_infer_acceleration/tests/generate_clip_score_datas.py

@@ -10,7 +10,7 @@
 from mindone.diffusers import StableDiffusion3Pipeline
 
 ms.set_context(deterministic="ON")
-ms.set_context(mode=0, jit_config={"jit_level": "O2"})
+ms.set_context(mode=0, jit_config={"jit_level": "O1"})
 
 
 def generate_imgs(pipe, save_path, use_cache_and_gate=True, use_todo=False):

examples/dit_infer_acceleration/tests/sd3_performance_test.py

@@ -51,7 +51,7 @@ def arg_parse():
 
 
 def main():
-    ms.set_context(mode=0, jit_config={"jit_level": "O2"})
+    ms.set_context(mode=0, jit_config={"jit_level": "O1"})
 
     args = arg_parse()
     current_path = os.path.abspath(os.path.dirname(__file__))