Commit e63400be authored by nanahira's avatar nanahira

Update Dockerfile.xformers

parent 9e65b5c6
Pipeline #17739 passed with stages
in 46 minutes and 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.gitmain#egg=xformers
pip install --no-cache-dir git+https://github.com/facebookresearch/xformers.git@main#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