diff --git a/environments/huggingface.yml b/environments/huggingface.yml index e053654a..fdf82e15 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -32,7 +32,7 @@ dependencies: - flask-ngrok - flask-cors - lupa==1.10 - - transformers==4.32.* + - transformers==4.32.1 - huggingface_hub==0.16.4 - optimum==1.12.0 - safetensors==0.3.3 diff --git a/environments/rocm.yml b/environments/rocm.yml index bd468a6a..00f50929 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -30,7 +30,7 @@ dependencies: - flask-ngrok - flask-cors - lupa==1.10 - - transformers==4.32.* + - transformers==4.32.1 - huggingface_hub==0.16.4 - optimum==1.12.0 - safetensors==0.3.3 diff --git a/requirements.txt b/requirements.txt index b7abbed1..323239c3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -transformers==4.32.* +transformers==4.32.1 huggingface_hub==0.16.4 optimum==1.12.0 safetensors==0.3.3