From 1ebea22857ce4948d46fe69f543067f4ab84047c Mon Sep 17 00:00:00 2001
From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com>
Date: Fri, 14 Apr 2023 18:25:38 +0200
Subject: [PATCH] fix(docker): fix image

---
 .github/workflows/build.yaml | 7 ++++---
 Dockerfile                   | 2 +-
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index cb51b273..91ebbd6b 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -79,9 +79,9 @@ jobs:
           flavor: |
             latest=auto
           images: |
-            ghcr.io/huggingface/text-generation-inference
+#            ghcr.io/huggingface/text-generation-inference
             registry.internal.huggingface.tech/api-inference/community/text-generation-inference
-            db4c2190dd824d1f950f5d1555fbadf0.azurecr.io/text-generation-inference
+#            db4c2190dd824d1f950f5d1555fbadf0.azurecr.io/text-generation-inference
           tags: |
             type=semver,pattern={{version}}
             type=semver,pattern={{major}}.{{minor}}
@@ -93,7 +93,8 @@ jobs:
         with:
           context: .
           file: Dockerfile
-          push: ${{ github.event_name != 'pull_request' }}
+#          push: ${{ github.event_name != 'pull_request' }}
+          push: true
           platforms: 'linux/amd64'
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
diff --git a/Dockerfile b/Dockerfile
index 62505f0c..c618a2f8 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -106,7 +106,7 @@ COPY server/Makefile-transformers Makefile
 RUN BUILD_EXTENSIONS="True" make build-transformers
 
 # Text Generation Inference base image
-FROM ubuntu:22.04 as base
+FROM debian:bullseye-slim as base
 
 ARG TARGETPLATFORM
 ARG PYTORCH_VERSION=2.0.0