Pin numpy<2 in vLLM image (#65)

modal-labs · Jun 18, 2024 · c942664 · c942664
1 parent f64c8d7
commit c942664
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/common.py b/src/common.py
@@ -40,6 +40,7 @@
 ).pip_install(
     "vllm==0.2.6",
     "torch==2.1.2",
+    "numpy<2",  # To avoid vLLM ecosystem compatibility issues
 )
 
 app = modal.App(