torch
torchdiffeq
torchsde
einops
open-clip-torch
transformers
safetensors
pytorch_lightning
aiohttp
accelerate
pyyaml
opencv-python
numpy
tqdm
torchvision
timm
scipy
Pillow
psutil
transformers