Skip to content

Commit 437a17e

Browse files
Merge branch 'int8' of https://github.com/TimDettmers/bitsandbytes into int8
2 parents 0460d2e + c7b31df commit 437a17e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bitsandbytes/research/autograd/_functions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ class SwitchBackBnb(torch.autograd.Function):
186186
@staticmethod
187187
# TODO: the B008 on the line below is a likely bug; the current implementation will
188188
# have each SwitchBackBnb instance share a single MatmulLtState instance!!!
189-
def forward(ctx, A, B, out=None, bias=None, state: MatmulLtState = None):
189+
def forward(ctx, A, B, out=None, bias=None, state: Optional[MatmulLtState] = None):
190190
state = state or MatmulLtState()
191191

192192
# default to pytorch behavior if inputs are empty

0 commit comments

Comments
 (0)