torch>=2.0.0
torchvision
transformers>=4.45.2
einops
open-clip-torch
tqdm
huggingface_hub
loguru
