unsloth/studio/backend/core/training
2026-07-27 03:40:17 +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 Invalidate latents on a VAE swap, keep a cut-off generation, surface the EMA adapter 2026-07-26 20:47:46 +00:00
diffusion_lora_trainer.py Trim the comments across the diffusion backend 2026-07-26 20:31:19 +00:00
diffusion_train_common.py Hide unloadable cached rows, hold the dataset interlock, bound a GIF export 2026-07-26 21:37:01 +00:00
diffusion_train_extras.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-07-27 03:40:17 +00:00
diffusion_training_service.py Hide unloadable cached rows, hold the dataset interlock, bound a GIF export 2026-07-26 21:37:01 +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 Trim the comments across the diffusion backend 2026-07-26 20:31:19 +00:00
worker.py AMD: CI coverage for recent fixes, plus three wrong gfx ids (#7431) 2026-07-25 18:58:02 -05:00