huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-10 11:54:52 +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: | flavor: |
latest=auto latest=auto
images: | images: |
ghcr.io/huggingface/text-generation-inference # ghcr.io/huggingface/text-generation-inference
registry.internal.huggingface.tech/api-inference/community/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: | tags: |
type=semver,pattern={{version}} type=semver,pattern={{version}}
type=semver,pattern={{major}}.{{minor}} type=semver,pattern={{major}}.{{minor}}
@ -93,7 +93,8 @@ jobs:
with: with:
context: . context: .
file: Dockerfile file: Dockerfile
push: ${{ github.event_name != 'pull_request' }} # push: ${{ github.event_name != 'pull_request' }}
push: true
platforms: 'linux/amd64' platforms: 'linux/amd64'
tags: ${{ steps.meta.outputs.tags }} tags: ${{ steps.meta.outputs.tags }}
labels: ${{ steps.meta.outputs.labels }} 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 RUN BUILD_EXTENSIONS="True" make build-transformers
# Text Generation Inference base image # Text Generation Inference base image
FROM ubuntu:22.04 as base FROM debian:bullseye-slim as base
ARG TARGETPLATFORM ARG TARGETPLATFORM
ARG PYTORCH_VERSION=2.0.0 ARG PYTORCH_VERSION=2.0.0