From c773e7833ec16000f8a72460a8aed2e9c8c71abc Mon Sep 17 00:00:00 2001 From: Michael Waskom Date: Tue, 18 Jun 2024 12:41:55 -0400 Subject: [PATCH] Pin numpy<2 in vLLM image --- src/common.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/common.py b/src/common.py index 64356fc..e1f9b9a 100644 --- 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(