Force compatible HF Hub

2022-10-23 18:28:50 +02:00 · 2022-10-23 18:28:50 +02:00 · 8ee795055c
parent 2795ced3a4
commit 8ee795055c
4 changed files with 4 additions and 0 deletions
--- a/environments/huggingface.yml
+++ b/environments/huggingface.yml
@ -25,4 +25,5 @@ dependencies:
    - flask-ngrok
    - lupa==1.10
    - transformers>=4.20.1
+    - huggingface_hub>=0.10.1
    - accelerate
--- a/environments/rocm.yml
+++ b/environments/rocm.yml
@ -25,4 +25,5 @@ dependencies:
    - flask-ngrok
    - lupa==1.10
    - transformers>=4.20.1
+    - huggingface_hub>=0.10.1
    - accelerate
--- a/requirements.txt
+++ b/requirements.txt
@ -1,4 +1,5 @@
 transformers>=4.20.1
+huggingface_hub>=0.10.1
 Flask
 Flask-SocketIO
 requests
--- 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