remove dlinfer version (#2672)

InternLM · Oct 28, 2024 · a07e65e · a07e65e
1 parent f5189ce
commit a07e65e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/Dockerfile_aarch64_ascend b/docker/Dockerfile_aarch64_ascend
@@ -110,7 +110,7 @@ RUN echo "source /usr/local/Ascend/ascend-toolkit/set_env.sh" >> ~/.bashrc && \
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip3 install torch==2.3.1 torchvision==0.18.1 torch-npu==2.3.1 && \
     pip3 install transformers timm && \
-    pip3 install dlinfer-ascend==0.1.1
+    pip3 install dlinfer-ascend
 
 # lmdeploy
 FROM build_temp as copy_temp