Skip to content

Commit 051311f

Browse files
authored
fix string (#18568)
1 parent 9a9a525 commit 051311f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/pipelines/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -768,7 +768,7 @@ def __init__(
768768
elif device < 0:
769769
self.device = torch.device("cpu")
770770
else:
771-
self.device = torch.device("cuda:{device}")
771+
self.device = torch.device(f"cuda:{device}")
772772
else:
773773
self.device = device
774774
self.binary_output = binary_output

0 commit comments

Comments
 (0)