mlflow==2.9.2
cloudpickle==2.2.1
datasets==2.15.0
openmim==0.3.9
torch==2.0.1
torchvision==0.15.2
transformers==4.36.2
accelerate==0.25.0
albumentations==1.3.0
fairscale==0.4.13
