numpy
diffusers==0.21.4
transformers
protobuf
sentencepiece
torch
torchvision
datasets
scipy
pytorch-msssim
matplotlib
huggingface-hub==0.25.2
opencv-python
scikit-image
bm3d
compressai