diff --git a/pyproject.toml b/pyproject.toml index e626cf2..49326ad 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -5,7 +5,7 @@ requires = [ build-backend = "hatchling.build" [project] name = "proof-wala" -version = "1.1.2" +version = "1.1.3" authors = [ { name="Amitayush Thakur", email="amitayush@utexas.edu" }, ] @@ -21,16 +21,16 @@ classifiers = [ dependencies = [ "itp-interface==1.1.2", "filelock==3.12.4", - "accelerate==0.23.0", + "accelerate==1.3.0", "bitsandbytes==0.41.1", "gradio==3.48.0", "protobuf==3.20.1", "scipy==1.11.2", "sentencepiece==0.1.99", "spaces==0.16.1", - "transformers==4.34.0", + "transformers==4.48.0", "peft==0.7.1", - "trl==0.7.7", + "trl==0.14.0", "parglare==0.16.1", "pydantic==1.10.13", "graphviz==0.17", diff --git a/requirements.txt b/requirements.txt index 6895056..657bc28 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ itp-interface==1.1.2 -accelerate==0.23.0 +accelerate==1.3.0 filelock==3.12.4 bitsandbytes==0.41.1 gradio==3.48.0 @@ -8,9 +8,9 @@ scipy==1.11.2 sentencepiece==0.1.99 # spaces==0.16.1 torch==2.1.1 -transformers==4.34.0 +transformers==4.48.0 peft==0.7.1 -trl==0.7.7 +trl==0.14.0 parglare==0.16.1 pydantic==1.10.13 # Bug with ray 2.4.0 where it is not compatible with pydantic >= 2 graphviz==0.17 diff --git a/tacc_requirements.txt b/tacc_requirements.txt index 5e5ec15..7e557cf 100644 --- a/tacc_requirements.txt +++ b/tacc_requirements.txt @@ -1,5 +1,5 @@ itp-interface==1.1.2 -accelerate==0.23.0 +accelerate==1.3.0 filelock==3.12.4 bitsandbytes==0.41.1 gradio==3.48.0 @@ -7,9 +7,9 @@ protobuf==3.20.3 scipy==1.11.2 sentencepiece==0.1.99 # spaces==0.16.1 -transformers==4.34.0 +transformers==4.48.0 peft==0.7.1 -trl==0.7.7 +trl==0.14.0 parglare==0.16.1 pydantic==1.10.13 # Bug with ray 2.4.0 where it is not compatible with pydantic >= 2 graphviz==0.17