From 4d8c8b20198f77e0cd5f3d8e6135df1aea0748c3 Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Sat, 27 Dec 2025 08:49:43 +0800
Subject: [PATCH] fix

---
 docker/docker-npu/Dockerfile         | 3 +++
 docker/docker-npu/docker-compose.yml | 1 -
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/docker/docker-npu/Dockerfile b/docker/docker-npu/Dockerfile
index c2f495588..70770c03f 100644
--- a/docker/docker-npu/Dockerfile
+++ b/docker/docker-npu/Dockerfile
@@ -40,6 +40,9 @@ RUN pip uninstall -y torch torchvision torchaudio && \
 # Set up volumes
 # VOLUME [ "/root/.cache/huggingface", "/app/shared_data", "/app/output" ]
 
+# Verify torch version
+RUN python -c "import torch, sys; v=torch.__version__; print('torch version:', v); sys.exit(0 if v.startswith('2.7.1') else 1)"
+
 # Expose port 7860 for LLaMA Board
 ENV GRADIO_SERVER_PORT=7860
 EXPOSE 7860
diff --git a/docker/docker-npu/docker-compose.yml b/docker/docker-npu/docker-compose.yml
index 621d7177e..8ef8ddd3b 100644
--- a/docker/docker-npu/docker-compose.yml
+++ b/docker/docker-npu/docker-compose.yml
@@ -35,7 +35,6 @@ services:
       args:
         BASE_IMAGE: quay.io/ascend/cann:8.3.rc2-a3-ubuntu22.04-py3.11
         PIP_INDEX: https://pypi.org/simple
-        EXTRAS: torch-npu,metrics
     container_name: llamafactory-a3
     image: llamafactory:npu-a3
     volumes: