From c7b31df34502d8c28a7e5d81f39ed85284a8930f Mon Sep 17 00:00:00 2001 From: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com> Date: Thu, 31 Oct 2024 11:40:47 -0400 Subject: [PATCH] Update bitsandbytes/research/autograd/_functions.py Co-authored-by: Aarni Koskela --- bitsandbytes/research/autograd/_functions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bitsandbytes/research/autograd/_functions.py b/bitsandbytes/research/autograd/_functions.py index f8fe14c48..5b31cca09 100644 --- a/bitsandbytes/research/autograd/_functions.py +++ b/bitsandbytes/research/autograd/_functions.py @@ -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