Commit 155f7b58 authored by nanahira's avatar nanahira

change xformers version

parent b34f2c88
Pipeline #17737 failed with stages
in 24 minutes and 44 seconds
......@@ -3,6 +3,6 @@ FROM ${BASE}
ARG GRAPHIC_VERSION
RUN env FORCE_CUDA=1 CUDA_VISIBLE_DEVICES=0 TORCH_CUDA_ARCH_LIST=${GRAPHIC_VERSION} \
pip install --no-cache-dir git+https://github.com/facebookresearch/xformers.git@v0.0.13#egg=xformers
pip install --no-cache-dir git+https://github.com/facebookresearch/xformers.git@v0.0.14#egg=xformers
CMD ["--listen", "--port", "8080", "--xformers"]
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