From ec7ea0b7607edeb082780386cad4dc9bc7798591 Mon Sep 17 00:00:00 2001
From: alpayariyak <aariyak@wpi.edu>
Date: Fri, 10 May 2024 00:00:48 +0000
Subject: [PATCH] Update default base image version to fix github actions build

---
 Dockerfile      | 2 +-
 docker-bake.hcl | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index f48b832..020f14e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
 ARG WORKER_CUDA_VERSION=11.8.0
-ARG BASE_IMAGE_VERSION=1.0.0
+ARG BASE_IMAGE_VERSION=1.0.0preview
 FROM runpod/worker-vllm:base-${BASE_IMAGE_VERSION}-cuda${WORKER_CUDA_VERSION} AS vllm-base
 
 RUN apt-get update -y \
diff --git a/docker-bake.hcl b/docker-bake.hcl
index f110a6d..fedfe81 100644
--- a/docker-bake.hcl
+++ b/docker-bake.hcl
@@ -43,7 +43,7 @@ target "base-1210" {
 }
 
 target "worker-1180" {
-  tags = ["${REPOSITORY}/worker-vllm:worker-${BASE_IMAGE_VERSION}-cuda11.8.0"]
+  tags = ["${REPOSITORY}/worker-vllm:${BASE_IMAGE_VERSION}-cuda11.8.0"]
   context = "."
   dockerfile = "Dockerfile"
   args = {
@@ -54,7 +54,7 @@ target "worker-1180" {
 }
 
 target "worker-1210" {
-  tags = ["${REPOSITORY}/worker-vllm:worker-${BASE_IMAGE_VERSION}-cuda12.1.0"]
+  tags = ["${REPOSITORY}/worker-vllm:${BASE_IMAGE_VERSION}-cuda12.1.0"]
   context = "."
   dockerfile = "Dockerfile"
   args = {