diff --git a/CHANGELOG.md b/CHANGELOG.md
index 1d42eee..1a66166 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -7,6 +7,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## Unreleased
 
+- Fixed torch version requirement to be compatible with flashinfer wheels (torch>=2.4.0,<2.5.0)
+
 ## [v0.1.58](https://github.com/allenai/olmocr/releases/tag/v0.1.58) - 2025-02-15
 
 ## [v0.1.53](https://github.com/allenai/olmocr/releases/tag/v0.1.53) - 2025-02-14
diff --git a/pyproject.toml b/pyproject.toml
index 51dd33d..2fa5ab6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -35,7 +35,7 @@ dependencies = [
   "zstandard",
   "boto3",
   "httpx",
-  "torch>=2.5.1",
+  "torch>=2.4.0,<2.5.0",
   "transformers>=4.46.2",
   "beaker-py",
 ]