From 6fa79050d32fa76c94739adca78221edc97f3abe Mon Sep 17 00:00:00 2001 From: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com> Date: Thu, 31 Oct 2024 10:45:07 -0400 Subject: [PATCH] Update bitsandbytes/functional.py Co-authored-by: Aarni Koskela --- bitsandbytes/functional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bitsandbytes/functional.py b/bitsandbytes/functional.py index d09463a64..d07162219 100644 --- a/bitsandbytes/functional.py +++ b/bitsandbytes/functional.py @@ -422,7 +422,7 @@ def get_special_format_str(): return "row" -def is_on_gpu(tensors: Iterable[torch.Tensor]): +def is_on_gpu(tensors: Iterable[Optional[torch.Tensor]]): on_gpu = True gpu_ids = set()