diff --git a/docker/docker-npu/Dockerfile b/docker/docker-npu/Dockerfile index ef4e2627c..f58c8d4d9 100644 --- a/docker/docker-npu/Dockerfile +++ b/docker/docker-npu/Dockerfile @@ -37,7 +37,7 @@ RUN pip uninstall -y torch torchvision torchaudio && \ COPY . /app # Install LLaMA Factory -RUN pip install --no-cache-dir -e ".[metrics,deepspeed]" --no-build-isolation +RUN pip install --no-cache-dir -e ".[metrics]" --no-build-isolation # Set up volumes # VOLUME [ "/root/.cache/huggingface", "/app/shared_data", "/app/output" ]