diff --git a/pyproject.toml b/pyproject.toml index 1d10d60b..725610e9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -11,11 +11,11 @@ version = "0.0.6" authors = [{ name = "Charles Goddard", email = "chargoddard@gmail.com" }] dependencies = [ "torch>=2.0.0", - "tqdm==4.66.5", - "click==8.1.7", - "safetensors~=0.4.3", - "accelerate~=1.0.1", - "pydantic~=2.9.2", + "tqdm==4.67.1", + "click==8.1.8", + "safetensors~=0.5.2", + "accelerate~=1.3.0", + "pydantic~=2.10.6", "immutables==0.20", "transformers>=4.45.2", "tokenizers>=0.20.1", @@ -29,8 +29,8 @@ dependencies = [ ] [project.optional-dependencies] -dev = ["black~=24.10.0", "isort~=5.13.2", "pre-commit~=4.0.1"] -test = ["pytest~=8.3.3"] +dev = ["black~=24.10.0", "isort~=5.13.2", "pre-commit~=4.1.0"] +test = ["pytest~=8.3.4"] evolve = ["ray", "cma", "lm_eval", "wandb"] vllm = ["vllm==0.3.2", "lm_eval[vllm]"]