Improve builds

roboflow · Feb 15, 2024 · 211c720 · 211c720
1 parent 79afc9d
commit 211c720
Show file tree

Hide file tree

Showing 6 changed files with 4 additions and 226 deletions.
diff --git a/.github/workflows/docker.cpu.yolo_world.yml b/.github/workflows/docker.cpu.yolo_world.yml
diff --git a/.github/workflows/docker.gpu.yolo_world.yml b/.github/workflows/docker.gpu.yolo_world.yml
diff --git a/docker/dockerfiles/Dockerfile.onnx.cpu b/docker/dockerfiles/Dockerfile.onnx.cpu
@@ -21,6 +21,7 @@ COPY requirements/requirements.sam.txt \
     requirements/requirements.gaze.txt \
     requirements/requirements.doctr.txt \
     requirements/requirements.groundingdino.txt \
+    requirements/requirements.yolo_world.txt \
     requirements/_requirements.txt \
     ./
 
@@ -34,6 +35,7 @@ RUN pip3 install --upgrade pip  && pip3 install \
     -r requirements.gaze.txt \
     -r requirements.doctr.txt \
     -r requirements.groundingdino.txt \
+    -r requirements.yolo_world.txt \
     jupyterlab \
     wheel>=0.38.0 \
     setuptools>=65.5.1 \

diff --git a/docker/dockerfiles/Dockerfile.onnx.cpu.yolo_world b/docker/dockerfiles/Dockerfile.onnx.cpu.yolo_world
diff --git a/docker/dockerfiles/Dockerfile.onnx.gpu b/docker/dockerfiles/Dockerfile.onnx.gpu
@@ -21,6 +21,7 @@ COPY requirements/requirements.sam.txt \
     requirements/requirements.doctr.txt \
     requirements/requirements.groundingdino.txt \
     requirements/requirements.cogvlm.txt \
+    requirements/requirements.yolo_world.txt \
     requirements/_requirements.txt \
     ./
 
@@ -35,6 +36,7 @@ RUN pip3 install --upgrade pip  && pip3 install \
     -r requirements.groundingdino.txt \
     -r requirements.doctr.txt \
     -r requirements.cogvlm.txt \
+    -r requirements.yolo_world.txt \
     jupyterlab \
     --upgrade \
     && rm -rf ~/.cache/pip

diff --git a/docker/dockerfiles/Dockerfile.onnx.gpu.yolo_world b/docker/dockerfiles/Dockerfile.onnx.gpu.yolo_world