fix: huggingface-hub requirements (#224)

instadeepai · Mar 5, 2024 · ed9a355 · ed9a355
1 parent a13f0bf
commit ed9a355
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/requirements/requirements-train.txt b/requirements/requirements-train.txt
@@ -1,5 +1,4 @@
 dm-haiku==0.0.9
-huggingface-hub
 hydra-core==1.3
 neptune-client==0.16.15
 optax>=0.1.4

diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -1,6 +1,7 @@
 chex>=0.1.3
 dm-env>=1.5
 gym>=0.22.0
+huggingface-hub
 jax>=0.2.26
 matplotlib~=3.7.4
 numpy>=1.19.5