34 lines
746 B
YAML
34 lines
746 B
YAML
# Copyright (C) 2024 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
tgi:
|
|
enabled: true
|
|
accelDevice: "gaudi"
|
|
image:
|
|
repository: ghcr.io/huggingface/tgi-gaudi
|
|
tag: "2.3.1"
|
|
resources:
|
|
limits:
|
|
habana.ai/gaudi: 1
|
|
MAX_INPUT_LENGTH: "1024"
|
|
MAX_TOTAL_TOKENS: "2048"
|
|
CUDA_GRAPHS: ""
|
|
OMPI_MCA_btl_vader_single_copy_mechanism: "none"
|
|
ENABLE_HPU_GRAPH: "true"
|
|
LIMIT_HPU_GRAPH: "true"
|
|
USE_FLASH_ATTENTION: "true"
|
|
FLASH_ATTENTION_RECOMPUTE: "true"
|
|
readinessProbe:
|
|
initialDelaySeconds: 5
|
|
periodSeconds: 5
|
|
timeoutSeconds: 1
|
|
startupProbe:
|
|
initialDelaySeconds: 5
|
|
periodSeconds: 5
|
|
timeoutSeconds: 1
|
|
failureThreshold: 120
|
|
vllm:
|
|
enabled: false
|
|
llm-uservice:
|
|
TEXTGEN_BACKEND: TGI
|