unsloth/studio/backend/core/training
Daniel Han 440bec0ec8 Tighten torchao configs and note the FSDP2 design for the DiT trainer
nf4 loads now enable double quantization (~0.4 bits/param off the frozen base
scales at no fidelity cost), fp8 training uses the rowwise recipe when the
torchao build ships it (per-row scaling confines the DiT activation outliers
that a tensor-wide scale collapses), and the inference quant filter gains a
per-scheme GEMM-tiling divisibility floor (16 for scaled_mm, 32 for MX blocks)
so one ragged Linear cannot crash the first denoise after a clean quantize
pass. plans/fsdp2_diffusion_design.md records the multi-GPU design: bf16/fp8
over FSDP2 with per-block units, LoRA attached before sharding, int8 out of
scope (DTensor over the quantized subclass is undefined), per-family notes.
2026-07-20 07:22:49 +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 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 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 Studio: auto-detect completion masking markers, stop silent full-sequence training (#7054) 2026-07-11 05:13:45 -07:00
training.py Tighten comments and docstrings added by the image-generation fixes 2026-07-13 05:29:09 +00:00
worker.py Studio: auto-detect completion masking markers, stop silent full-sequence training (#7054) 2026-07-11 05:13:45 -07:00