unsloth/studio/backend/core/training
2026-07-26 14:46:53 +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 [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-07-26 12:40:35 +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 [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-07-26 14:46:53 +00:00
diffusion_train_extras.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-07-26 12:40:35 +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 Fix training start NameError, the load-order guard test and CPU-only diffusion tests 2026-07-25 19:18:35 -07:00
worker.py AMD: CI coverage for recent fixes, plus three wrong gfx ids (#7431) 2026-07-25 18:58:02 -05:00