fix string (#18568)

051311ff · Michael Wyatt · GitHub · 9a9a525b · 051311ff
Unverified Commit 051311ff authored Aug 10, 2022 by Michael Wyatt Committed by GitHub Aug 10, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/pipelines/base.py src/transformers/pipelines/base.py +1 -1

No files found.
--- a/src/transformers/pipelines/base.py
+++ b/src/transformers/pipelines/base.py
@@ -768,7 +768,7 @@ class Pipeline(_ScikitCompat):
            elif device < 0:
                self.device = torch.device("cpu")
            else:
-                self.device = torch.device("cuda:{device}")
+                self.device = torch.device(f"cuda:{device}")
        else:
            self.device = device
        self.binary_output = binary_output