fix string (#18568)

huggingface · Aug 10, 2022 · 051311f · 051311f
1 parent 9a9a525
commit 051311f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/pipelines/base.py b/src/transformers/pipelines/base.py
@@ -768,7 +768,7 @@ def __init__(
             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