Restore is_torch_greater_or_equal_than for backward compatibility (#3…

…5734) * Restore is_torch_greater_or_equal_than for backward compatibility Signed-off-by: Tyler Michael Smith <[email protected]> * review comments Signed-off-by: Tyler Michael Smith <[email protected]> --------- Signed-off-by: Tyler Michael Smith <[email protected]>
huggingface · Jan 30, 2025 · f7b6047 · f7b6047
1 parent 2e752ea
commit f7b6047
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/transformers/pytorch_utils.py b/src/transformers/pytorch_utils.py
@@ -35,6 +35,11 @@
 is_torch_greater_or_equal_than_2_2 = parsed_torch_version_base >= version.parse("2.2")
 is_torch_greater_or_equal_than_2_1 = parsed_torch_version_base >= version.parse("2.1")
 
+# For backwards compatibility (e.g. some remote codes on Hub using those variables).
+is_torch_greater_or_equal_than_2_0 = parsed_torch_version_base >= version.parse("2.0")
+is_torch_greater_or_equal_than_1_13 = parsed_torch_version_base >= version.parse("1.13")
+is_torch_greater_or_equal_than_1_12 = parsed_torch_version_base >= version.parse("1.12")
+
 # Cache this result has it's a C FFI call which can be pretty time-consuming
 _torch_distributed_available = torch.distributed.is_available()