requirements.txt

git+https://github.com/huggingface/diffusers.git@integrations/cosmos-predict2
transformers
accelerate
sentencepiece
safetensors
torchvision
git+https://github.com/yiyixuxu/cosmos-guardrail.git
peft