Skip to content

Commit e39b304

Browse files
MekkCyberzucchini-nlp
authored andcommitted
Make sure torch_is_available before using torch.distributed (huggingface#37693)
fix
1 parent 0395675 commit e39b304

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/pipelines/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -997,7 +997,7 @@ def __init__(
997997
else:
998998
self.device = device if device is not None else -1
999999

1000-
if torch.distributed.is_initialized():
1000+
if is_torch_available() and torch.distributed.is_initialized():
10011001
self.device = self.model.device
10021002
logger.warning(f"Device set to use {self.device}")
10031003

0 commit comments

Comments
 (0)