diff --git a/setup.py b/setup.py
index cc3037ebb7..968881c56c 100644
--- a/setup.py
+++ b/setup.py
@@ -685,7 +685,7 @@ def _read_requirements(filename: str) -> list[str]:
                   "mistral_common[audio]"],  # Required for audio processing
         "video": [],  # Kept for backwards compatibility
         # FlashInfer should be updated together with the Dockerfile
-        "flashinfer": ["flashinfer-python==0.2.11"],
+        "flashinfer": ["flashinfer-python==0.6.14"],
     },
     cmdclass=cmdclass,
     package_data=package_data,