Skip to content

Commit

Permalink
Merge branch 'int8' of https://github.com/TimDettmers/bitsandbytes in…
Browse files Browse the repository at this point in the history
…to int8
  • Loading branch information
matthewdouglas committed Nov 4, 2024
2 parents 0460d2e + c7b31df commit 437a17e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion bitsandbytes/research/autograd/_functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -186,7 +186,7 @@ class SwitchBackBnb(torch.autograd.Function):
@staticmethod
# TODO: the B008 on the line below is a likely bug; the current implementation will
# have each SwitchBackBnb instance share a single MatmulLtState instance!!!
def forward(ctx, A, B, out=None, bias=None, state: MatmulLtState = None):
def forward(ctx, A, B, out=None, bias=None, state: Optional[MatmulLtState] = None):
state = state or MatmulLtState()

# default to pytorch behavior if inputs are empty
Expand Down

0 comments on commit 437a17e

Please sign in to comment.