From 6dacc7c9ba5696d1cb5ebaaa68e0ec64e242bc88 Mon Sep 17 00:00:00 2001 From: Michael Wyatt Date: Wed, 10 Aug 2022 14:28:27 -0700 Subject: [PATCH] fix string --- src/transformers/pipelines/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/pipelines/base.py b/src/transformers/pipelines/base.py index a0ce06ec5e33..7842b95b3285 100644 --- 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