torch==2.1.0
torchvision==0.16.0
timm==0.9.12
diffusers==0.26.3
accelerate==0.21.0
onnx==1.15.0
onnxsim==0.4.36
numpy==1.24.4
protobuf==3.20.2
pytorch-fid