fix bug

huggingface · Mar 8, 2024 · f761c88 · f761c88
1 parent 0d218e2
commit f761c88
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/utils/import_utils.py b/src/transformers/utils/import_utils.py
@@ -539,7 +539,7 @@ def is_torch_xla_available(check_is_tpu=False, check_is_gpu=False):
 @lru_cache()
 def is_torch_neuroncore_available(check_device=True):
     if importlib.util.find_spec("torch_neuronx") is not None:
-        return is_torch_xla_available(check_device)
+        return is_torch_xla_available()
     return False