fix string (huggingface#18568)

amyeroberts · Oct 18, 2022 · e97f675 · e97f675
1 parent 0c0220a
commit e97f675
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