xformers
xformers copied to clipboard
could you support for nvidia pytorch docker images?
Nvidia pytorch docker images is very easy to use, but it haven't a suitable version for xformers. could you support for nvidia pytorch docker images?
https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/index.html
you can do it yourself, refs: https://github.com/facebookresearch/xformers/issues/960