unsloth/studio/backend/core/training
Daniel Han 76520bb553 Retain diffusion training loss history and expose it in status
The training service kept only the latest loss, so a live loss chart could show a
single point. Fold each progress event into bounded (step, loss, lr) history arrays
(capped at 4000 points, decimated when full) plus the latest throughput and peak VRAM,
and record the family / base model / catalog path on completion. The status endpoint
returns these as a nested metric_history object the UI can chart directly, and the
start request accepts an optional model_family override.
2026-07-02 14:55:17 +00:00
..
__init__.py Reduce and tighten code comments and docstrings repo-wide (#6095) 2026-06-08 23:09:51 -07:00
diffusion_lora_trainer.py Refactor diffusion LoRA training into a family-aware platform 2026-07-02 14:55:09 +00:00
diffusion_train_common.py Refactor diffusion LoRA training into a family-aware platform 2026-07-02 14:55:09 +00:00
diffusion_training_service.py Retain diffusion training loss history and expose it in status 2026-07-02 14:55:17 +00:00
resume.py feat(studio): implement S3 dataset loading (completes #5951) (#6222) 2026-06-12 14:52:04 +02:00
s3_dataset.py feat(studio): implement S3 dataset loading (completes #5951) (#6222) 2026-06-12 14:52:04 +02:00
trainer.py Fix Windows Studio UTF-8 startup handling (#6614) 2026-07-01 13:47:33 +01:00
training.py (feat) Add project names to studio training runs (#6512) 2026-06-29 16:06:36 +02:00
worker.py (feat) Add project names to studio training runs (#6512) 2026-06-29 16:06:36 +02:00