requirements.txt

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