Add hf_transfer to our dependencies (#32)

Add hf_transfer to the dependencies to enable `HF_HUB_ENABLE_HF_TRANSFER=1` feature for faster model downloading.
marqo-ai · Dec 9, 2024 · f822301 · f822301
1 parent b0d5b22
commit f822301
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 5 deletions.
diff --git a/requirements/amd64-gpu-requirements.txt b/requirements/amd64-gpu-requirements.txt
@@ -6,7 +6,7 @@
 #
 --extra-index-url https://download.pytorch.org/whl/cu113
 
-aiohappyeyeballs==2.4.3
+aiohappyeyeballs==2.4.4
     # via aiohttp
 aiohttp==3.10.9
     # via
@@ -494,4 +494,7 @@ zipp==3.20.2
     # via
     #   -r requirements.in
     #   importlib-metadata
-    #   importlib-resources
+    #   importlib-resources
+hf-transfer==0.1.8
+    # via
+    #   -r requirements.in
diff --git a/requirements/arm64-requirements.txt b/requirements/arm64-requirements.txt
@@ -6,7 +6,7 @@
 #
 --extra-index-url https://download.pytorch.org/whl/cu113
 
-aiohappyeyeballs==2.4.3
+aiohappyeyeballs==2.4.4
     # via aiohttp
 aiohttp==3.10.9
     # via
@@ -481,4 +481,7 @@ zipp==3.20.2
     # via
     #   -r requirements.in
     #   importlib-metadata
-    #   importlib-resources
+    #   importlib-resources
+hf-transfer==0.1.8
+    # via
+    #   -r requirements.in
diff --git a/requirements/requirements.in b/requirements/requirements.in
@@ -96,4 +96,5 @@ watchfiles==0.24.0
 websockets==13.1
 zipp==3.20.2
 yarl==1.13.1
-portalocker==2.10.1
+portalocker==2.10.1
+hf-transfer==0.1.8