Skip to content

Commit 6fa7905

Browse files
matthewdouglasakx
andauthored
Update bitsandbytes/functional.py
Co-authored-by: Aarni Koskela <[email protected]>
1 parent b9cb5c9 commit 6fa7905

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bitsandbytes/functional.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -422,7 +422,7 @@ def get_special_format_str():
422422
return "row"
423423

424424

425-
def is_on_gpu(tensors: Iterable[torch.Tensor]):
425+
def is_on_gpu(tensors: Iterable[Optional[torch.Tensor]]):
426426
on_gpu = True
427427
gpu_ids = set()
428428

0 commit comments

Comments
 (0)