riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Make sure torch_is_available before using torch.distributed (#37693)

Browse Source 
fix
This commit is contained in:
Mohamed Mekkouri 2025-04-24 11:31:35 +02:00 committed by GitHub
parent 864e9636ff
commit 02baa61fab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/pipelines/base.py
View File

@ -997,7 +997,7 @@ class Pipeline(_ScikitCompat, PushToHubMixin):
else:
self.device = device if device is not None else -1
if torch.distributed.is_initialized():
if is_torch_available() and torch.distributed.is_initialized():
self.device = self.model.device
logger.warning(f"Device set to use {self.device}")