Daniel Han d6795ed077 Restore pre-Ampere bf16 fail-fast in the DiT trainer ... The perf rewrite dropped the bf16 capability guard, so a pre-Ampere CUDA device (T4/V100/RTX 20xx) would die deep in model load with an opaque dtype error instead of a clear message. Restores parity with the SDXL trainer.		2026-07-05 05:27:58 +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	Restore pre-Ampere bf16 fail-fast in the DiT trainer	2026-07-05 05:27:58 +00:00
diffusion_lora_trainer.py	Merge image-generation: grad norm chart, completion state, Windows caption keys, GGUF compute copy	2026-07-04 04:33:30 +00:00
diffusion_train_common.py	Merge branch 'image-generation' into diffusion-train-perf	2026-07-05 02:09:39 +00:00
diffusion_training_service.py	[pre-commit.ci] auto fixes from pre-commit.com hooks	2026-07-04 05:08:43 +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