pip install git+https://github.com/huggingface/diffusers.git@integrations/cosmos-predict2
transformers
accelerate
sentencepiece
safetensors
torchvision
git+https://github.com/a-r-r-o-w/cosmos-guardrail.git@patch-1
peft