transformers==4.37
torch==2.1.0 
torchvision==0.16.0 
torchaudio==2.1.0
mlflow
boto3
pytest
pydantic
datasets
psutil
sagemaker
gradio
waitress