Force compatible HF Hub

henk717 · henk717 · commit 8ee795055cbe · 2022-10-23T18:28:50.000+02:00
diff --git 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
diff --git 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
diff --git 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
@@ -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
