Skip to content

Commit b573920

Browse files
committed
[TRTLLM-6607] fix NSPECT issue
Signed-off-by: ZhanruiSunCh <184402041+ZhanruiSunCh@users.noreply.github.com>
1 parent 450ebe3 commit b573920

File tree

4 files changed

+2
-18
lines changed

4 files changed

+2
-18
lines changed

constraints.txt

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,2 @@
11
# These vulnerabilities were inherited from the base image (pytorch:25.06-py3) and should be removed when the base image
22
# is updated.
3-
4-
# WAR against https://github.com/advisories/GHSA-8qvm-5x2c-j2w7
5-
protobuf>=4.25.8

docker/Dockerfile.multi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ RUN GITHUB_MIRROR=${GITHUB_MIRROR} bash ./install.sh --mpi4py && rm install_mpi4
7171
ARG TORCH_INSTALL_TYPE="skip"
7272
RUN TORCH_INSTALL_TYPE=${TORCH_INSTALL_TYPE} bash ./install.sh --pytorch && rm install_pytorch.sh
7373

74-
RUN bash ./install.sh --opencv && bash ./install.sh --protobuf && rm install.sh
74+
RUN bash ./install.sh --opencv && rm install.sh
7575

7676
FROM ${TRITON_IMAGE}:${TRITON_BASE_TAG} AS triton
7777

docker/common/install.sh

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ polygraphy=0
1616
mpi4py=0
1717
pytorch=0
1818
opencv=0
19-
protobuf=0
2019

2120
while [[ $# -gt 0 ]]; do
2221
case $1 in
@@ -56,10 +55,6 @@ while [[ $# -gt 0 ]]; do
5655
opencv=1
5756
shift 1
5857
;;
59-
--protobuf)
60-
protobuf=1
61-
shift 1
62-
;;
6358
--all)
6459
base=1
6560
cmake=1
@@ -70,7 +65,6 @@ while [[ $# -gt 0 ]]; do
7065
mpi4py=1
7166
pytorch=1
7267
opencv=1
73-
protobuf=1
7468
shift 1
7569
;;
7670
*)
@@ -135,10 +129,3 @@ if [ $opencv -eq 1 ]; then
135129
rm -rf /usr/local/lib/python3*/dist-packages/cv2/
136130
pip3 install opencv-python-headless --force-reinstall --no-deps --no-cache-dir
137131
fi
138-
139-
# WARs against security issues inherited from pytorch:25.06
140-
# * https://github.com/advisories/GHSA-8qvm-5x2c-j2w7
141-
if [ $protobuf -eq 1 ]; then
142-
pip3 install --upgrade --no-cache-dir \
143-
"protobuf>=4.25.8"
144-
fi

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ prometheus_fastapi_instrumentator
3333
pydantic>=2.9.1
3434
pydantic-settings[yaml]
3535
omegaconf
36-
pillow==10.3.0
36+
pillow
3737
wheel<=0.45.1
3838
optimum
3939
# evaluate needs datasets>=2.0.0 which triggers datasets>3.1.0 which is not stable: https://github.com/huggingface/datasets/issues/7467

0 commit comments

Comments
 (0)