riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-04 05:10:06 +06:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

guard torch distributed check (#39057)

Browse Source 
* guard torch distributed check

* Update src/transformers/pipelines/base.py

---------

Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
This commit is contained in:
Tijana Vukovic 2025-06-27 16:49:47 +02:00 committed by GitHub
parent 49d9fd49bd
commit c8764ab935
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

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