diff --git a/environments/huggingface.yml b/environments/huggingface.yml index d65f9446..205d5e31 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -20,5 +20,5 @@ dependencies: - flask-cloudflared - flask-ngrok - lupa==1.10 - - transformers>=4.20 + - transformers>=4.20.1 - accelerate \ No newline at end of file diff --git a/environments/rocm.yml b/environments/rocm.yml index d3e10515..8ade341f 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -20,5 +20,5 @@ dependencies: - flask-cloudflared - flask-ngrok - lupa==1.10 - - transformers>=4.20 + - transformers>=4.20.1 - accelerate diff --git a/requirements.txt b/requirements.txt index e6308790..13fdd7cd 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -transformers>=4.20 +transformers>=4.20.1 Flask Flask-SocketIO requests