From ed9a355da56495bc2a8377e4cc6924f0ff0ffcfa Mon Sep 17 00:00:00 2001 From: Sasha Date: Tue, 5 Mar 2024 11:06:14 +0200 Subject: [PATCH] fix: huggingface-hub requirements (#224) --- requirements/requirements-train.txt | 1 - requirements/requirements.txt | 1 + 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements/requirements-train.txt b/requirements/requirements-train.txt index 644e7f1bd..087ca0369 100644 --- 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 index 4e43bdd4e..2e398c025 100644 --- 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