Fix: I'm a Python beginner

2023-12-30 20:07:43 +09:00
parent 9045fa3426
commit 4faef57dd2
1 changed files with 1 additions and 1 deletions
--- 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)