Commit Graph

7 Commits

Author SHA1 Message Date
Christian Medina
75aeeeb20a fix: use torchrun for distributed training, load model on CPU 2026-06-30 18:45:14 -04:00
Christian Medina
5ad5518ca6 fix: add deepspeed to auto-install dependencies 2026-06-30 18:30:23 -04:00
Christian Medina
3d87f0b8b3 fix: force reinstall PyTorch to resolve version conflicts 2026-06-30 17:58:52 -04:00
Christian Medina
7e8ddabf03 fix: remove dataset download, it's already in repo 2026-06-30 17:11:29 -04:00
Christian Medina
28d0a956fd fix: update PyTorch to cu124 for RTX 5090, fix total_memory typo 2026-06-30 16:58:11 -04:00
Christian Medina
1786830d84 feat: unload AI model before training 2026-06-30 16:00:17 -04:00
Christian Medina
44939e374b refactor: simplify deploy script, add train-on-this-server.sh 2026-06-30 15:22:54 -04:00