File tree Expand file tree Collapse file tree 2 files changed +3
-4
lines changed Expand file tree Collapse file tree 2 files changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -42,9 +42,9 @@ RUN pip install "nvidia-modelopt[all]" -U
4242ENV 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"
4343RUN 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
4646COPY . 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
Original file line number Diff line number Diff line change 66sentencepiece >= 0.2.0
77tensorboardX
88torch >= 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
You can’t perform that action at this time.
0 commit comments