huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-09 19:34:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(docker): fix image

Browse Source
This commit is contained in:
OlivierDehaene 2023-04-14 18:25:38 +02:00
parent 1bb394631d
commit 1ebea22857
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
.github/workflows/build.yaml vendored
View File

@ -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 }}

2
Dockerfile
View File

@ -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