axolotl-ai-cloud / axolotl

Go ahead and axolotl questions

https://axolotl-ai-cloud.github.io/axolotl/

Apache License 2.0

7.73k stars 851 forks source link

bump xformers to 0.0.27 #1740

Closed akshaylive closed 3 months ago

akshaylive commented 3 months ago

Description

Update xformers version.

Motivation and Context

Preserve compatibility with torch 2.3.1. Reference

How has this been tested?

 pip3 install -e '.[flash-attn,deepspeed]'

winglian commented 3 months ago

so we need to cross check the wheels they are building for the new version and make sure they didn't drop pytorch/cuda version for some wheels. I think this is likely fine, but when we update xformers in requirements, we parse it in setup.py so we can replace older versions based on the installed version of python in case there isn't a wheel for it.

winglian commented 3 months ago

Thanks @akshaylive!