torch
einops
accelerate==0.34.2
einops==0.8.0
transformers==4.41.2
huggingface-hub==0.24.6
datasets
omegaconf
diffusers
sentencepiece
opencv-python
matplotlib
onnxruntime
torchvision
timm
invisible-watermark
fire
tqdm