From 4faef57dd24d445ff232b8a7df42f69e7738e233 Mon Sep 17 00:00:00 2001 From: litagin02 Date: Sat, 30 Dec 2023 20:07:43 +0900 Subject: [PATCH] Fix: I'm a Python beginner --- text/english_bert_mock.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/text/english_bert_mock.py b/text/english_bert_mock.py index 328adff..feee598 100644 --- a/text/english_bert_mock.py +++ b/text/english_bert_mock.py @@ -28,7 +28,7 @@ def get_bert_feature( device = "mps" if not device: device = "cuda" - if device = "cuda" and not torch.cuda.is_available(): + if device == "cuda" and not torch.cuda.is_available(): device = "cpu" if device not in models.keys(): models[device] = DebertaV2Model.from_pretrained(LOCAL_PATH).to(device)