Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "42f63e3871ea10bb9dfc9345642bceb36458ac62"
Unverified Commit 051311ff authored by Michael Wyatt's avatar Michael Wyatt Committed by GitHub
Browse files

fix string (#18568)

parent 9a9a525b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/pipelines/base.py
View file @ 051311ff
...@@ -768,7 +768,7 @@ class Pipeline(_ScikitCompat): ...@@ -768,7 +768,7 @@ class Pipeline(_ScikitCompat):
elif device < 0: elif device < 0:
self.device = torch.device("cpu") self.device = torch.device("cpu")
else: else:
self.device = torch.device("cuda:{device}") self.device = torch.device(f"cuda:{device}")
else: else:
self.device = device self.device = device
self.binary_output = binary_output self.binary_output = binary_output
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment