Daniel Han 90a79843a3 diffusion: gate SDXL bf16 fallback on native bf16 (compute-capability probe) ... torch.cuda.is_bf16_supported() reports True on pre-Ampere GPUs that only emulate bf16, so the SDXL LoRA trainer would keep bf16 there and fail at load/forward. Use native_bf16_supported() (the same compute-capability probe the DiT trainer already uses) so T4 / V100 / RTX 20xx fall back to fp16 instead.		2026-07-09 06:45:00 +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	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