Commit 7c3da5c0 authored by helloyongyang's avatar helloyongyang
Browse files

update transformers version

parent 8013b8a2
...@@ -25,7 +25,11 @@ RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 1 ...@@ -25,7 +25,11 @@ RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 1
&& update-alternatives --install /usr/bin/python python /usr/bin/python3.11 1 && update-alternatives --install /usr/bin/python python /usr/bin/python3.11 1
RUN pip config set global.index-url https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple \ RUN pip config set global.index-url https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple \
&& pip install -r /workspace/lightx2v/requirements.txt --use-deprecated=legacy-resolver && pip install -r /workspace/lightx2v/requirements.txt
# Install again separately to bypass the version conflict check
RUN pip config set global.index-url https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple \
&& pip install transformers==4.45.2
# install flash-attention 2 # install flash-attention 2
RUN cd lightx2v/3rd/flash-attention && pip install --no-cache-dir -v -e . RUN cd lightx2v/3rd/flash-attention && pip install --no-cache-dir -v -e .
......
...@@ -27,7 +27,10 @@ git submodule update --init --recursive ...@@ -27,7 +27,10 @@ git submodule update --init --recursive
# create conda env and install requirments # create conda env and install requirments
conda create -n lightx2v python=3.11 && conda activate lightx2v conda create -n lightx2v python=3.11 && conda activate lightx2v
pip install -r requirements.txt --use-deprecated=legacy-resolver pip install -r requirements.txt
# Install again separately to bypass the version conflict check
pip install transformers==4.45.2
# install flash-attention 2 # install flash-attention 2
cd lightx2v/3rd/flash-attention && pip install --no-cache-dir -v -e . cd lightx2v/3rd/flash-attention && pip install --no-cache-dir -v -e .
......
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