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