remove need for manual git clone

Author: mnoukhov

Dockerfile

@@ -72,9 +72,6 @@ ENV UV_COMPILE_BYTECODE=0
 # Copy only dependency-related files first
 COPY pyproject.toml uv.lock ./
 
-# Install custom vllm for olmo3
-RUN git clone -b shanea/olmo3 https://github.com/2015aroras/vllm.git vllm_olmo3
-
 # Install dependencies
 RUN --mount=type=cache,target=${UV_CACHE_DIR} \
     --mount=type=bind,source=uv.lock,target=uv.lock \

pyproject.toml

@@ -24,7 +24,7 @@ dependencies = [
     "tensorboard>=2.18.0",
     "torch>=2.7.0,<2.8",
     "transformers @ git+https://github.com/2015aroras/transformers.git@shanea/olmo3",
-    "vllm",
+    "vllm @ git+https://github.com/2015aroras/vllm.git@shanea/olmo3",
     "wandb==0.18.1",
     "langdetect==1.0.9",
     "immutabledict==1.2.0",
@@ -53,7 +53,6 @@ vllm = { VLLM_USE_PRECOMPILED = "1" }
 torch = [
   { index = "pytorch-cu128", marker = "platform_system != 'Darwin'"},
 ]
-vllm = { path = "./vllm_olmo3", editable = true }
 
 [[tool.uv.index]]
 name = "pytorch-cu128"