Files
kubeflow-pipelines/components/runpod_trainer/requirements.txt
Greg Hendrickson 4ff491f847 feat: Use self-hosted runner + curated DDI dataset
- Switch to self-hosted runner on compute-01 for faster builds
- Replace PyTDC with curated DDI dataset (no heavy deps)
- 60+ real drug interaction patterns based on clinical guidelines
- Generates up to 10K training samples with text variations
- Maintains 5-level severity classification
2026-02-03 03:27:10 +00:00

10 lines
155 B
Plaintext

runpod>=1.7.0
transformers==4.44.0
datasets>=2.16.0
accelerate>=0.30.0
boto3>=1.34.0
scikit-learn>=1.3.0
scipy>=1.11.0
safetensors>=0.4.0
requests>=2.31.0