Merge pull request #473 from MeTaNoV/patch-1

Update flag_models.py to fix is_torch_npu_available as a function call
FlagOpen · Feb 21, 2024 · af538b6 · af538b6
2 parents 9365f15 + 3c9ab5a
commit af538b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/FlagEmbedding/flag_models.py b/FlagEmbedding/flag_models.py
@@ -26,7 +26,7 @@ def __init__(
             self.device = torch.device("cuda")
         elif torch.backends.mps.is_available():
             self.device = torch.device("mps")
-        elif is_torch_npu_available:
+        elif is_torch_npu_available():
             self.device = torch.device("npu")
         else:
             self.device = torch.device("cpu")