a92f18bf71
* krea2: don't hardcode NVIDIA-only cuDNN SDPA backend The krea2 attention() forced SDPBackend.CUDNN_ATTENTION, which is NVIDIA-only. On non-NVIDIA backends (AMD ROCm, Intel XPU, Apple MPS) every forward pass fails with 'RuntimeError: No available kernel. Aborting execution.', so Krea 2 LoRA training cannot run at all there. Pass a priority list [CUDNN, FLASH, EFFICIENT, MATH] instead. NVIDIA still selects cuDNN; other backends fall back to flash/efficient/math. Verified training end-to-end on an AMD Radeon 8060S (gfx1151, ROCm 7.2). * Version bump * Add set priority flag so CUDNN_ATTENTION is selected on cuda devices first. --------- Co-authored-by: Jaret Burkett <jaretburkett@gmail.com>
2 lines
20 B
Python
2 lines
20 B
Python
VERSION = "0.10.25"
|