[safetensors] don't use in `torch<1.10` (#22370)

* [safetensors] don't use in pt<1.10 * better fix

[safetensors] don't use in `torch<1.10` (#22370)
* [safetensors] don't use in pt<1.10 * better fix
cae78c46 · Stas Bekman · GitHub · cfab34e1 · cae78c46
Unverified Commit cae78c46 authored Mar 24, 2023 by Stas Bekman Committed by GitHub Mar 24, 2023
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 1 deletion

src/transformers/utils/import_utils.py src/transformers/utils/import_utils.py +7 -1

No files found.
--- a/src/transformers/utils/import_utils.py
+++ b/src/transformers/utils/import_utils.py
@@ -577,6 +577,12 @@ def is_optimum_available():


 def is_safetensors_available():
+    if is_torch_available():
+        if version.parse(_torch_version) >= version.parse("1.10"):
+            return importlib.util.find_spec("safetensors") is not None
+        else:
+            return False
+    else:
        return importlib.util.find_spec("safetensors") is not None