From 09541947b80181803e8b8a4ba9d96620d36e4371 Mon Sep 17 00:00:00 2001 From: Aarni Koskela Date: Mon, 29 Jan 2024 15:18:28 +0200 Subject: [PATCH] Fix missed call to is_triton_available --- bitsandbytes/nn/triton_based_modules.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bitsandbytes/nn/triton_based_modules.py b/bitsandbytes/nn/triton_based_modules.py index de07ac647..67b45f4a5 100644 --- a/bitsandbytes/nn/triton_based_modules.py +++ b/bitsandbytes/nn/triton_based_modules.py @@ -162,7 +162,7 @@ def __init__( ): super().__init__(in_features, out_features, bias, device, dtype) - if not is_triton_available: + if not is_triton_available(): raise ImportError('''Could not import triton. Please install triton to use SwitchBackLinear. Alternatively, you can use bnb.nn.SwitchBackLinearBnb, but it will be slower''')