Skip to content

Commit 0ecb8fb

Browse files
committed
lint
1 parent 2938c73 commit 0ecb8fb

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

bitsandbytes/nn/modules.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -360,12 +360,12 @@ def to(self, *args, **kwargs):
360360
def __torch_function__(cls, func, types, args=(), kwargs=None):
361361
if kwargs is None:
362362
kwargs = {}
363-
363+
364364
if func in [torch.chunk, torch.split]:
365365
tensor = args[0]
366-
366+
367367
result = super().__torch_function__(func, types, args, kwargs)
368-
368+
369369
if isinstance(result, tuple):
370370
return tuple(
371371
cls(
@@ -393,9 +393,9 @@ def __torch_function__(cls, func, types, args=(), kwargs=None):
393393
module=tensor.module,
394394
bnb_quantized=tensor.bnb_quantized,
395395
)
396-
396+
397397
return super().__torch_function__(func, types, args, kwargs)
398-
398+
399399

400400
def fix_4bit_weight_quant_state_from_module(module: Union["Embedding4bit", "Linear4bit"]):
401401
if getattr(module.weight, "quant_state", None) is not None:

0 commit comments

Comments
 (0)