Fix 2609 - Import packaging by default

huggingface · Oct 16, 2024 · b3ceb9d · b3ceb9d
1 parent 9bb6bd6
commit b3ceb9d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/huggingface_hub/hub_mixin.py b/src/huggingface_hub/hub_mixin.py
@@ -17,6 +17,8 @@
     Union,
 )
 
+import packaging.version
+
 from . import constants
 from .errors import EntryNotFoundError, HfHubHTTPError
 from .file_download import hf_hub_download
@@ -41,7 +43,6 @@
     import torch  # type: ignore
 
 if is_safetensors_available():
-    import packaging.version
     import safetensors
     from safetensors.torch import load_model as load_model_as_safetensor
     from safetensors.torch import save_model as save_model_as_safetensor