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. |
||
|---|---|---|
| .. | ||
| backend | ||
| frontend | ||
| src-tauri | ||
| __init__.py | ||
| install_llama_prebuilt.py | ||
| install_node_prebuilt.py | ||
| install_python_stack.py | ||
| install_sd_cpp_prebuilt.py | ||
| LICENSE.AGPL-3.0 | ||
| node_prebuilt_pins.json | ||
| package-lock.json | ||
| package.json | ||
| setup.bat | ||
| setup.ps1 | ||
| setup.sh | ||
| Unsloth_Studio_Colab.ipynb | ||