Commit 1eb888af authored by nanahira's avatar nanahira

lock xformers version

parent 46646a8a
Pipeline #24791 passed with stages
in 46 minutes and 38 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@main#egg=xformers
pip install --no-cache-dir git+https://github.com/facebookresearch/xformers.git@v0.0.20#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