diff --git a/Dockerfile b/Dockerfile
index dbf3e1d30..0a4a86315 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -5,7 +5,6 @@ FROM ${BASE_IMAGE}
 ARG ONEFLOW_PIP_INDEX
 ARG ONEFLOW_PACKAGE_NAME=oneflow
 RUN pip install -f ${ONEFLOW_PIP_INDEX} ${ONEFLOW_PACKAGE_NAME}
+RUN python3 -m pip install "torch" "transformers==4.27.1" "diffusers[torch]==0.19.3"
 ADD . /src/onediff
-RUN python3 -m pip install "torch" "transformers==4.27.1" "diffusers[torch]==0.19.3" && \
-    python3 -m pip uninstall accelerate -y && \
-    python3 -m pip install -e /src/onediff
+RUN python3 -m pip install -e /src/onediff
diff --git a/tests/comfy-docker-compose.yml b/tests/comfy-docker-compose.yml
index 9ded08296..e5e4c6578 100644
--- a/tests/comfy-docker-compose.yml
+++ b/tests/comfy-docker-compose.yml
@@ -25,6 +25,8 @@ services:
     environment:
       ONEFLOW_RUN_GRAPH_BY_VM: "0"
       HF_HUB_OFFLINE: "1"
+      ONEFLOW_MLIR_ENABLE_TIMING: "1"
+      ONEFLOW_MLIR_PRINT_STATS: "1"
       CI: "1"
       SILICON_ONEDIFF_LICENSE_KEY: ${SILICON_ONEDIFF_LICENSE_KEY}
     volumes:
diff --git a/tests/diffusers-docker-compose.yml b/tests/diffusers-docker-compose.yml
index 09429c454..ccf94ad73 100644
--- a/tests/diffusers-docker-compose.yml
+++ b/tests/diffusers-docker-compose.yml
@@ -15,6 +15,8 @@ services:
       - seccomp=unconfined
     environment:
       HF_HUB_OFFLINE: "1"
+      ONEFLOW_MLIR_ENABLE_TIMING: "1"
+      ONEFLOW_MLIR_PRINT_STATS: "1"
       ONEFLOW_RUN_GRAPH_BY_VM: "0"
       CI: "1"
       SILICON_ONEDIFF_LICENSE_KEY: ${SILICON_ONEDIFF_LICENSE_KEY}