Commit Graph

15 Commits

Author SHA1 Message Date
39922e8d2e feat: Add Gemma 3 12B with QLoRA fine-tuning
- Added PEFT, bitsandbytes, TRL for LoRA training
- 4-bit QLoRA quantization for 48GB GPU fit
- Instruction-tuning format for Gemma chat template
- Auto-detect model type (BERT vs LLM)
- Updated GPU tier to ADA_24/AMPERE_48
2026-02-03 03:58:25 +00:00
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
afc8fc6690 feat: Add real DrugBank DDI dataset support via TDC
- Added PyTDC dependency for DrugBank access
- Implemented DDI type -> severity label mapping (0-4)
- Added train/eval split with stratification
- Added accuracy and F1 metrics for evaluation
- Default: 50K samples from DrugBank DDI
- Supports both real data and custom inline data
2026-02-03 02:48:31 +00:00
0f4858d22f fix: disable checkpoint saving to avoid tensor contiguity error 2026-02-03 02:38:15 +00:00
2680ad5502 fix: remove MinIO dependency, use inline training data 2026-02-03 02:27:49 +00:00
b086239c52 Fix Python/transformers version compatibility 2026-02-03 01:11:20 +00:00
297a75da4d Remove internal domains from README 2026-02-03 00:45:27 +00:00
c459bb3139 Add GHA workflow to build DDI trainer image 2026-02-03 00:28:09 +00:00
222be0fb68 Use Tailscale endpoints, add RunPod Docker build files 2026-02-03 00:23:16 +00:00
07bb8aa6bb Fix MinIO endpoint to use internal cluster service 2026-02-03 00:15:26 +00:00
9ca3d6c195 Add DDI training pipeline with RunPod serverless GPU support 2026-02-02 23:56:05 +00:00
11da494a4f Use self-hosted runner on compute-01 for full GitOps 2026-02-02 23:46:14 +00:00
db0bd65281 Add GHA workflow for pipeline compilation 2026-02-02 23:41:49 +00:00
09ee583d6e Add .gitignore to prevent credential leaks 2026-02-02 23:39:06 +00:00
591a312399 Initial Kubeflow GitOps setup with example pipelines 2026-02-02 23:37:53 +00:00