torch
torchvision
torchaudio
diffusers
transformers
numpy
loguru
sentencepiece
tqdm