unsloth/studio/backend/core/training
Daniel Han 5f65f01e2e Fix training start blocking, dataset resolution shadowing, duplicate uploads, partial-caption gate
Run backend.start_training off the event loop with asyncio.to_thread so the
synchronous diffusion/video unload calls (which wait on engine generation
locks) cannot freeze concurrent requests; guard against overlapping starts
with a _start_in_progress compare-and-set under the service lock.

Resolve bare diffusion dataset names directly under datasets_root() before
falling back to the generic resolver, so an unrelated LLM upload file or
recipe folder sharing the name cannot shadow the image dataset.

Reject exact duplicate filenames within one multipart upload batch: two
parts staged to the same destination would let the later tmp.replace
silently discard the earlier file. Case variants stay exempt per the
existing stem-guard contract.

Require an instance prompt in the train panel when only some images have
captions, since backend discovery silently skips uncaptioned images.
2026-07-10 06:56:32 +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 Gate DiT-training bf16 on native compute capability, not emulated is_bf16_supported 2026-07-08 05:00:18 +00:00
diffusion_lora_trainer.py diffusion: gate SDXL bf16 fallback on native bf16 (compute-capability probe) 2026-07-09 06:45:00 +00:00
diffusion_train_common.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-07-08 05:02:13 +00:00
diffusion_training_service.py Make the diffusion-training reservation a compare-and-set 2026-07-07 11:57:37 +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 Add MLX backend support for CLI unsloth train (#6709) 2026-07-08 03:25:26 -07:00
training.py Fix training start blocking, dataset resolution shadowing, duplicate uploads, partial-caption gate 2026-07-10 06:56:32 +00:00
worker.py Add MLX backend support for CLI unsloth train (#6709) 2026-07-08 03:25:26 -07:00