Commit 9e65b5c6 authored by nanahira's avatar nanahira

Update Dockerfile.xformers

parent 155f7b58
Pipeline #17738 canceled with stages
in 13 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.14#egg=xformers
pip install --no-cache-dir git+https://github.com/facebookresearch/xformers.gitmain#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