diff --git a/pyproject.toml b/pyproject.toml index ede9e0d8bf..05f8269991 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -175,8 +175,9 @@ huggingfacenotorch = [ ] amd = [ "unsloth[huggingfacenotorch]", - # bnb <= 0.49.2 NaNs at decode shape on every AMD GPU; the ROCm 4-bit GEMV - # fix (bnb PR #1887) first ships in 0.50.0, which is now on PyPI. + # 4-bit decode is unreliable on ROCm before 0.50.0, the first PyPI release + # carrying the full path: blocksize/warp decoupling (bnb #1887), fused SIMT + # GEMM on RDNA (#1979), RDNA3/4 workgroup fix (#2012). "bitsandbytes>=0.50.0 ; ('linux' in sys_platform) and (platform_machine == 'AMD64' or platform_machine == 'x86_64' or platform_machine == 'aarch64')", "bitsandbytes>=0.50.0 ; (sys_platform == 'win32') and (platform_machine == 'AMD64' or platform_machine == 'x86_64')", ]