Revert "Update flag_models.py to fix is_torch_npu_available as a func…

…tion call"
Joecsr24 · Feb 21, 2024 · 2eb4360 · 2eb4360
1 parent af538b6
commit 2eb4360
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")