[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
1de299c673
commit
aff62c8591
1 changed files with 1 additions and 3 deletions
|
|
@ -2032,9 +2032,7 @@ def _ensure_rocm_torch() -> None:
|
|||
)
|
||||
if not _bnb_installed:
|
||||
_fallback_note = (
|
||||
", which carries the ROCm 4-bit fix"
|
||||
if _bnb_rocm_arch_has_binary()
|
||||
else ""
|
||||
", which carries the ROCm 4-bit fix" if _bnb_rocm_arch_has_binary() else ""
|
||||
)
|
||||
print(
|
||||
_red(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue