Update bitsandbytes/functional.py

Co-authored-by: Aarni Koskela <[email protected]>
bitsandbytes-foundation · Oct 31, 2024 · 6fa7905 · 6fa7905
1 parent b9cb5c9
commit 6fa7905
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()