unsloth/studio/backend/core/training
michaelhan bfe6f542ce Merge origin/main into image-generation (PR #6763)
Resolve the drift between PR #6763 and current main:

- deletion: main moved cached-model deletion into hub/services/models/deletion.py,
  so the PR's Images/Video in-use guards move there too as _diffusion_blocks_delete
  and _video_blocks_delete, keeping main's fail-closed 503 contract.
- llama_keepwarm: take main's rewrite, re-apply the PR's image/video inference
  suffixes so a generation in flight blocks an idle unload.
- routes/training: keep main's sidecar-swap 409 and resume_source_run_id, run
  start_training in the worker thread the PR's unload hook needs.
- model picker: main renamed components/assistant-ui/model-selector ->
  features/model-picker/... and rewrote pickers.tsx, so the PR's picker work is
  ported onto main's version (task/catalog props, task gating of hub + cached +
  local rows, single-device expanderGpuGb, fine-tuned section hidden when scoped)
  rather than reverting main's pinned-models and per-model-config work.
- images/video pages: imports repointed at the new model-selector path.
- tests: delete-guard tests retargeted at the deletion service.

Typecheck, i18n parity and model-catalog checks pass.
2026-07-25 00:34:38 -07: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 Add LoRA EMA, a persistent conditioning cache, and aspect bucketing helpers 2026-07-20 07:21:24 +00:00
diffusion_lora_trainer.py Tighten comments across the remaining image stack files 2026-07-12 11:46:23 +00:00
diffusion_train_common.py Add LoRA EMA, a persistent conditioning cache, and aspect bucketing helpers 2026-07-20 07:21:24 +00:00
diffusion_train_extras.py Add LoRA EMA, a persistent conditioning cache, and aspect bucketing helpers 2026-07-20 07:21:24 +00:00
diffusion_training_service.py Tighten comments across the remaining image stack files 2026-07-12 11:46:23 +00:00
fsdp2_design_notes.md Tighten torchao configs and note the FSDP2 design for the DiT trainer 2026-07-20 07:22:49 +00:00
resume.py Fix resume training crash recovery and MLX checkpoints (#6796) 2026-07-21 02:34:58 -07:00
s3_dataset.py feat(studio): implement S3 dataset loading (completes #5951) (#6222) 2026-06-12 14:52:04 +02:00
trainer.py feat(studio): add DoRA support to studio (#7315) 2026-07-24 03:24:16 -07:00
training.py Merge origin/main into image-generation (PR #6763) 2026-07-25 00:34:38 -07:00
worker.py feat(studio): add DoRA support to studio (#7315) 2026-07-24 03:24:16 -07:00