From 9167c8a1b87f070bb6b2c029403e787c5af2f65d Mon Sep 17 00:00:00 2001 From: Dushyant Behl Date: Fri, 28 Nov 2025 14:30:13 +0530 Subject: [PATCH] fix nvcr dockerfile build Signed-off-by: Dushyant Behl --- build/nvcr.Dockerfile | 5 +++-- pyproject.toml | 1 + 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/build/nvcr.Dockerfile b/build/nvcr.Dockerfile index e277b7f25b..1ee94d5e2e 100644 --- a/build/nvcr.Dockerfile +++ b/build/nvcr.Dockerfile @@ -47,8 +47,9 @@ RUN pip install --upgrade --force-reinstall torch torchaudio torchvision --index # Install main package + flash attention COPY . ${SOURCE_DIR} RUN cd ${SOURCE_DIR} -RUN pip install --no-cache-dir ${SOURCE_DIR} && \ - pip install --no-cache-dir ${SOURCE_DIR}[flash-attn] + +RUN pip install --no-cache-dir ${SOURCE_DIR} +RUN pip install --user --no-build-isolation ${SOURCE_DIR}[flash-attn] # Optional extras RUN if [[ "${ENABLE_FMS_ACCELERATION}" == "true" ]]; then \ diff --git a/pyproject.toml b/pyproject.toml index 7b772f2d4a..928ce8fcb0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -30,6 +30,7 @@ dependencies = [ "accelerate>=1.9.0,<2.0.0", "transformers>=4.55.0,<=4.55.4", "torch>2.7.0,<2.9.0", +"torchvision<0.24", "sentencepiece>=0.1.99,<0.3", "tokenizers<=0.22", "tqdm>=4.66.2,<5.0",