lru restore

sambhavnoobcoder · Jan 23, 2025 · 393869b · 393869b
1 parent 689b39f
commit 393869b
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
@@ -645,7 +645,7 @@ def is_g2p_en_available():
     return _g2p_en_available
 
 
-@lru_cache
+@lru_cache()
 def is_torch_tpu_available(check_device=True):
     "Checks if `torch_xla` is installed and potentially if a TPU is in the environment"
     warnings.warn(