unsloth/studio/backend/core/training
Daniel Han 123b3ea4d6 Make the diffusion-training reservation a compare-and-set
Two overlapping /diffusion/start requests can interleave between the is_active()
check and the reservation, so reserve() itself must reject a second reservation
atomically. Otherwise both callers reserve, both free the GPU's resident chat or
image model, and the loser only 409s after the eviction -- the evict-then-fail the
reservation exists to prevent. reserve() now raises under the lock if a start is
already reserved or a job is already running.
2026-07-07 11:57:37 +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 Merge remote-tracking branch 'origin/diffusion-train-perf2' into fold-integration 2026-07-07 01:06:42 +00:00
diffusion_lora_trainer.py Merge branch 'diffusion-krea2' into diffusion-train-perf2 2026-07-05 07:56:51 +00:00
diffusion_train_common.py Reject doomed local picks before the GPU handoff; load bare local safetensors 2026-07-07 08:06:54 +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 Fix Windows Studio UTF-8 startup handling (#6614) 2026-07-01 13:47:33 +01:00
training.py (feat) Add project names to studio training runs (#6512) 2026-06-29 16:06:36 +02:00
worker.py Add MLX-aware public Unsloth trainer API (#6462) 2026-07-02 23:02:26 +01:00