Skip to content

Commit

Permalink
Fix k8s container BASE_IMAGE (#2808)
Browse files Browse the repository at this point in the history
docker/build_image.sh creates a CPU image with tag 'latest-cpu'.  Use
that tag when building the k8s kserve image.

Co-authored-by: Sean Bowman <r.bowman@striveworks.us>
Co-authored-by: lxning <23464292+lxning@users.noreply.github.com>
  • Loading branch information
3 people committed Nov 28, 2023
1 parent 7baa8a6 commit 709e743
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion kubernetes/kserve/build_image.sh
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

MACHINE=cpu
DOCKER_TAG="pytorch/torchserve-kfs:latest"
BASE_IMAGE="pytorch/torchserve:latest"
BASE_IMAGE="pytorch/torchserve:latest-cpu"
DOCKER_FILE="Dockerfile"
BUILD_NIGHTLY=false
USE_CUSTOM_TAG=false
Expand Down

0 comments on commit 709e743

Please sign in to comment.