Skip to content

GitLab

"python/vscode:/vscode.git/clone" did not exist on "0b3a5b115115e7cab60d62035d9382c228beea87"
Unverified Commit adfd81d3 authored by lvhan028's avatar lvhan028 Committed by GitHub
Browse files

lower transformer version <4.30.0 (#66)

parent 3de27ead
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
docker/Dockerfile
View file @ adfd81d3
...@@ -5,6 +5,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ ...@@ -5,6 +5,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
RUN python3 -m pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117 RUN python3 -m pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
RUN python3 -m pip install sentencepiece cmake transformers protobuf==3.20.3 pybind11 mmengine RUN python3 -m pip install sentencepiece cmake transformers<4.30.0 protobuf==3.20.3 pybind11 mmengine
ENV NCCL_LAUNCH_MODE=GROUP ENV NCCL_LAUNCH_MODE=GROUP
requirements.txt
View file @ adfd81d3
...@@ -3,4 +3,5 @@ mmengine ...@@ -3,4 +3,5 @@ mmengine
numpy numpy
safetensors safetensors
setuptools setuptools
transformers<4.30.0
tritonclient==2.33.0 tritonclient==2.33.0
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment