Skip to content

Commit

Permalink
Fix missed call to is_triton_available
Browse files Browse the repository at this point in the history
  • Loading branch information
akx committed Jan 29, 2024
1 parent e3e01a8 commit 0954194
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion bitsandbytes/nn/triton_based_modules.py
Original file line number Diff line number Diff line change
Expand Up @@ -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''')

Expand Down

0 comments on commit 0954194

Please sign in to comment.