diff --git a/pyproject.toml b/pyproject.toml index 84babdbf..1901edd5 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -40,7 +40,7 @@ dependencies = [ "pybind11", "ray", "tensordict", - "transformers<4.48", + "transformers<4.52", "vllm<=0.6.3", ] diff --git a/requirements.txt b/requirements.txt index 2bf1d121..cec5b2be 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,7 +9,7 @@ pandas pybind11 ray tensordict<0.6 -transformers<4.48 +transformers<4.52 vllm<=0.6.3 wandb IPython