[Docs] xformers support

@DavidDragonsage

Testing by @DavidDragonsage here confirms that xformers installation is possible on supported platforms known to torchruntime.

The only known supported platform at the moment is cu124. However, rocm6.2.4 support will also be available if #9 is accepted.

It would be helpful if this extra feature is documented. The additional package can be installed from the command line with:

torchruntime install xformers

It can also be installed using the API with:

import torchruntime
torchruntime.install(["xformers"])

ROCm 6.3 also has an xformers build.

As of 12/27, the following Torch platforms have xformers support:

Notes:

See list here; 6.2.4 is supported but not enabled ([Feature] ROCm 6.2.4 support #9)
External to the Torch ecosystem

Navigation Menu