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

trigger build

Browse Source
This commit is contained in:
OlivierDehaene 2023-04-04 15:32:52 +02:00
parent f9b09d9629
commit 8604d37015
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -65,20 +65,20 @@ 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}}
type=raw,value=latest,enable=${{ github.ref == format('refs/heads/{0}', github.event.repository.default_branch) }} # type=raw,value=latest,enable=${{ github.ref == format('refs/heads/{0}', github.event.repository.default_branch) }}
type=raw,value=sha-${{ env.GITHUB_SHA_SHORT }} type=raw,value=llama-sha-${{ env.GITHUB_SHA_SHORT }}
- name: Build and push Docker image - name: Build and push Docker image
uses: docker/build-push-action@v2 uses: docker/build-push-action@v2
with: with:
context: . context: .
file: Dockerfile file: Dockerfile
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
server/text_generation_server/models/model.py
View File

@ -32,7 +32,7 @@ class Model(ABC):
skip_special_tokens=False, skip_special_tokens=False,
) )
if results[0][0] == " " and results[1][0] != " ": if results[0] and results[0][0] == " " and results[1][0] != " ":
results[0] = results[0].lstrip() results[0] = results[0].lstrip()
# slice to remove previous token # slice to remove previous token