90a79843a3
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. |
||
|---|---|---|
| .. | ||
| assets | ||
| auth | ||
| core | ||
| hub | ||
| loggers | ||
| models | ||
| plugins | ||
| requirements | ||
| routes | ||
| state | ||
| storage | ||
| tests | ||
| utils | ||
| __init__.py | ||
| _platform_compat.py | ||
| cloudflare_tunnel.py | ||
| colab.py | ||
| main.py | ||
| run.py | ||
| startup_banner.py | ||