Files
kubeflow-pipelines/components/runpod_trainer/requirements.txt

10 lines
151 B
Plaintext

runpod>=1.6.0
transformers>=4.36.0
datasets>=2.16.0
accelerate>=0.25.0
boto3>=1.34.0
scikit-learn>=1.3.0
scipy>=1.11.0
torch>=2.1.0
safetensors>=0.4.0