diff --git a/pyproject.toml b/pyproject.toml index 6301986..23e0504 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -38,7 +38,7 @@ dependencies = [ "boto3", "httpx", "torch>=2.7.0", - "transformers==4.57.3", + "transformers==5.0.0rc3", "beaker-py", ] license = {file = "LICENSE"} @@ -51,7 +51,7 @@ Changelog = "https://github.com/allenai/olmocr/blob/main/CHANGELOG.md" [project.optional-dependencies] gpu = [ - "vllm==0.11.2" + "vllm==0.20.0" ] dev = [