Daniel Han b8da49118d Diffusion LoRA training: fall back to fp16 when CUDA lacks bf16 ... The default mixed_precision=bf16 hard-fails on pre-Ampere GPUs (T4 / V100 / RTX 20xx) which have no bf16 compute; check torch.cuda.is_bf16_supported() and drop to fp16 there.		2026-07-01 23:55:41 +00:00
..
__init__.py	Reduce and tighten code comments and docstrings repo-wide (#6095)	2026-06-08 23:09:51 -07:00
diffusion_lora_trainer.py	Diffusion LoRA training: fall back to fp16 when CUDA lacks bf16	2026-07-01 23:55:41 +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	Studio: fix misleading "increase max_seq_length" message for train-on-completions (#6664)	2026-06-25 03:30:12 -07:00
training.py	(feat) Add project names to studio training runs (#6512)	2026-06-29 16:06:36 +02:00
worker.py	(feat) Add project names to studio training runs (#6512)	2026-06-29 16:06:36 +02:00