diff --git a/setup.py b/setup.py
index 77d74ed4..5ea1fa3a 100644
--- a/setup.py
+++ b/setup.py
@@ -42,7 +42,7 @@
     "accelerate",
     "datasets>=2.20",
     "zstandard",
-    "huggingface_hub @ git+https://github.com/huggingface/huggingface_hub@fix-discard-shared-tensors",
+    "huggingface_hub @ git+https://github.com/huggingface/huggingface_hub",
 ]
 
 setup(