diff --git a/requirements.txt b/requirements.txt index 7f5a952..611b62f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -63,5 +63,5 @@ sentencepiece google protobuf ninja -transformers==4.36.2 +transformers==5.0.0 pyobjc-framework-Metal \ No newline at end of file diff --git a/testing_requirements.txt b/testing_requirements.txt index a3969a8..fcc718a 100644 --- a/testing_requirements.txt +++ b/testing_requirements.txt @@ -61,6 +61,6 @@ ray sentencepiece google protobuf -transformers==4.36.2 +transformers==5.0.0 triton \ No newline at end of file