diff --git a/src/onediff/infer_compiler/oneflow_compiler_config.py b/src/onediff/infer_compiler/oneflow_compiler_config.py
index e0963ec8c..c1d969190 100644
--- a/src/onediff/infer_compiler/oneflow_compiler_config.py
+++ b/src/onediff/infer_compiler/oneflow_compiler_config.py
@@ -11,8 +11,8 @@
 
 def init_default_env():
     # ONEFLOW_RUN_GRAPH_BY_VM must set here to enable nn.Graph init with vm run
-    # os.environ.setdefault("ONEFLOW_RUN_GRAPH_BY_VM", "1")
-    # os.environ.setdefault("ONEFLOW_GRAPH_DELAY_VARIABLE_OP_EXECUTION", "1")
+    os.environ.setdefault("ONEFLOW_RUN_GRAPH_BY_VM", "1")
+    os.environ.setdefault("ONEFLOW_GRAPH_DELAY_VARIABLE_OP_EXECUTION", "1")
 
     os.environ.setdefault("ONEFLOW_MLIR_CSE", "1")
     os.environ.setdefault("ONEFLOW_MLIR_ENABLE_INFERENCE_OPTIMIZATION", "1")