From 8ee795055cbe0ba27c0812181e498a92efd39fd4 Mon Sep 17 00:00:00 2001 From: Henk Date: Sun, 23 Oct 2022 18:28:50 +0200 Subject: [PATCH] Force compatible HF Hub --- environments/huggingface.yml | 1 + environments/rocm.yml | 1 + requirements.txt | 1 + requirements_mtj.txt | 1 + 4 files changed, 4 insertions(+) diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 7abceefa..6654b470 100644 --- 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 index e885f4df..87fac8bb 100644 --- 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 index b1e2247c..9412771d 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