SwissArmyTransformer>=0.4.12 omegaconf>=2.3.0 pytorch_lightning>=2.4.0 kornia>=0.7.3 beartype>=0.19.0 fsspec>=2024.2.0 safetensors>=0.4.5 scipy>=1.14.1 decord>=0.6.0 wandb>=0.18.5 deepspeed>=0.15.3