unsloth/studio/backend/core/training
Daniel Han bfbb902610 Fix review findings: engine unload before training, caption precedence, dataset caption counts, local diffusers tagging, family LoRA targets
- Unload the ACTIVE image engine (sd_cpp or diffusers) before diffusion training starts, not just the diffusers singleton
- Count metadata.jsonl captions in dataset summaries so metadata-captioned datasets are not reported as uncaptioned
- Sidecar captions now override metadata rows everywhere (grid edits win); trainer and dataset API agree
- Tag local diffusers image checkpoints with text-to-image so they appear in the Images picker
- Family LoRA targets (_FLUX_TARGETS etc) apply when the config carries the generic defaults; explicit overrides still win
2026-07-04 03:23:05 +00:00
..
__init__.py Reduce and tighten code comments and docstrings repo-wide (#6095) 2026-06-08 23:09:51 -07:00
diffusion_dit_trainer.py Fix review findings: engine unload before training, caption precedence, dataset caption counts, local diffusers tagging, family LoRA targets 2026-07-04 03:23:05 +00:00
diffusion_lora_trainer.py FLUX.1-dev, Qwen-Image, and Z-Image LoRA training (#6820) 2026-07-03 16:19:16 -03:00
diffusion_train_common.py Fix review findings: engine unload before training, caption precedence, dataset caption counts, local diffusers tagging, family LoRA targets 2026-07-04 03:23:05 +00:00
diffusion_training_service.py Diffusion training platform: family registry, metric history, adapter metadata (#6819) 2026-07-03 16:18:01 -03: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 Add MLX-aware public Unsloth trainer API (#6462) 2026-07-02 23:02:26 +01:00