mlflow>=2.0.0
PyYAML>=6.0
scikit-learn>=1.2.0
transformers>=4.30.0
torch>=2.0.0
peft>=0.4.0
psutil
pynvml
