diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 9957f1c5..c1a168ae 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -26,5 +26,6 @@ dependencies: - flask-ngrok - lupa==1.10 - transformers>=4.20.1 + - huggingface_hub>=0.10.1 - accelerate - git+https://github.com/VE-FORBRYDERNE/mkultra diff --git a/environments/rocm.yml b/environments/rocm.yml index c6fea0b3..03425cf8 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -26,5 +26,6 @@ dependencies: - flask-ngrok - lupa==1.10 - transformers>=4.20.1 + - huggingface_hub>=0.10.1 - accelerate - git+https://github.com/VE-FORBRYDERNE/mkultra diff --git a/requirements.txt b/requirements.txt index 0d2bcd6b..355a458c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ transformers>=4.20.1 +huggingface_hub>=0.10.1 Flask Flask-SocketIO requests diff --git a/requirements_mtj.txt b/requirements_mtj.txt index 743c9c1d..90011fda 100644 --- a/requirements_mtj.txt +++ b/requirements_mtj.txt @@ -6,6 +6,7 @@ dm-haiku == 0.0.5 jax == 0.2.21 jaxlib >= 0.1.69, <= 0.3.7 transformers >= 4.20.1 +huggingface_hub >= 0.10.1 progressbar2 git+https://github.com/VE-FORBRYDERNE/mesh-transformer-jax@ck flask