torch
transformers
timm
einops
datasets
bitsandbytes
accelerate
peft
open_clip_torch
lovely_tensors
lion-pytorch
pytorch_optimizer
onnxsim
git+https://github.com/apple/ml-mobileclip.git