torch torchvision packaging ninja # flash-attn --no-build-isolation -U xformers --index-url https://download.pytorch.org/whl/cu121 -e git+https://github.com/hpcaitech/Open-Sora.git#egg=open-sora