diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 77f239b4..a841e630 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -23,6 +23,6 @@ dependencies: - flask-cloudflared - flask-ngrok - lupa==1.10 - - transformers>=4.20.1 + - transformers==4.21.3 - accelerate - - loguru + - loguru \ No newline at end of file diff --git a/environments/rocm.yml b/environments/rocm.yml index 142d4230..5aee0dca 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -23,5 +23,5 @@ dependencies: - flask-cloudflared - flask-ngrok - lupa==1.10 - - transformers>=4.20.1 + - transformers==4.21.3 - accelerate diff --git a/requirements.txt b/requirements.txt index 4923a140..260858ee 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -transformers>=4.20.1 +transformers==4.21.3 Flask Flask-SocketIO requests diff --git a/requirements_mtj.txt b/requirements_mtj.txt index ebdd5d98..cfbb7cad 100644 --- a/requirements_mtj.txt +++ b/requirements_mtj.txt @@ -6,7 +6,7 @@ optax >= 0.0.5, <= 0.0.9 dm-haiku == 0.0.5 jax == 0.2.21 jaxlib >= 0.1.69, <= 0.3.7 -transformers >= 4.19 +transformers == 4.21.3 progressbar2 git+https://github.com/VE-FORBRYDERNE/mesh-transformer-jax@ck flask