transformers>=4.40
torch
torchvision
safetensors
einops
ptwt
wandb
peft