accelerate
beartype
classifier-free-guidance-pytorch>=0.1.4
einops>=0.7.0
ema-pytorch
torch>=2.0
torch_geometric
torchtyping
vector-quantize-pytorch>=1.11.8
x-transformers>=1.26.0
tqdm
