From ed56977a8c4b850d984517bc2da29319d20cc4c3 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 18 Jul 2024 14:43:03 -0700 Subject: [PATCH] Update __init__.py --- unsloth/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/__init__.py b/unsloth/__init__.py index 6a2d999b41..4640681543 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -89,7 +89,7 @@ if (major_torch == 2) and (minor_torch >= 5): return old_is_bf16_supported(including_emulation) torch.cuda.is_bf16_supported = is_bf16_supported else: - def is_bf16_supported(): SUPPORTS_BFLOAT16 + def is_bf16_supported(): return SUPPORTS_BFLOAT16 torch.cuda.is_bf16_supported = is_bf16_supported pass