From b9cb5c93f88fcba1c62ce7af3ae09eb6dcb81197 Mon Sep 17 00:00:00 2001 From: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com> Date: Thu, 31 Oct 2024 10:39:03 -0400 Subject: [PATCH] Update bitsandbytes/autograd/_functions.py Co-authored-by: Aarni Koskela --- bitsandbytes/autograd/_functions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bitsandbytes/autograd/_functions.py b/bitsandbytes/autograd/_functions.py index 5d9983545..d0963a1e9 100644 --- a/bitsandbytes/autograd/_functions.py +++ b/bitsandbytes/autograd/_functions.py @@ -291,7 +291,7 @@ def forward( B: torch.Tensor, out=None, bias: Optional[torch.Tensor] = None, - state: MatmulLtState = None, + state: Optional[MatmulLtState] = None, ): state = state or MatmulLtState()