Skip to content

Commit eee97da

Browse files
Fix docker installation to skip windows requirements
1 parent f77cae9 commit eee97da

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

docker/Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,9 +42,9 @@ RUN pip install "nvidia-modelopt[all]" -U
4242
ENV TORCH_CUDA_ARCH_LIST="5.2 6.0 6.1 7.0 7.2 7.5 8.0 8.6 8.7 9.0+PTX"
4343
RUN python -c "import modelopt.torch.quantization.extensions as ext; ext.precompile()"
4444

45-
# Find and install requirements.txt files for all examples
45+
# Find and install requirements.txt files for all examples excluding windows
4646
COPY . TensorRT-Model-Optimizer
47-
RUN find TensorRT-Model-Optimizer -name "requirements*.txt" | while read req_file; do \
47+
RUN find TensorRT-Model-Optimizer -name "requirements.txt" | grep -v "windows" | while read req_file; do \
4848
echo "Installing from $req_file"; \
4949
pip install -r "$req_file"; \
5050
done

llm_qat/requirements.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,5 +6,4 @@ py7zr
66
sentencepiece>=0.2.0
77
tensorboardX
88
torch>=2.1.0
9-
# fix the version of transformers due to the issue https://github.com/huggingface/transformers/issues/33413
10-
transformers>=4.44.0,<=4.44.2
9+
transformers>=4.47.0

0 commit comments

Comments
 (0)