Este tutorial descreve como implementar um modelo de linguagem (conteúdo extenso) (MDL/CE) no Google Kubernetes Engine (GKE) com o GKE Inference Gateway. O tutorial inclui passos para a configuração do cluster, a implementação do modelo, a configuração do GKE Inference Gateway e o processamento de pedidos de MDIs.
Este tutorial destina-se a engenheiros de aprendizagem automática (ML), administradores e operadores de plataformas, e especialistas em dados e IA que querem implementar e gerir aplicações de MDI/CE no GKE com o GKE Inference Gateway.
Antes de ler esta página, familiarize-se com o seguinte:
- Acerca da inferência de modelos no GKE
- Execute a inferência de práticas recomendadas com as receitas de início rápido da inferência do GKE
- Modo Autopilot e modo Standard
- GPUs no GKE
Contexto
Esta secção descreve as principais tecnologias usadas neste tutorial. Para mais informações acerca dos conceitos e da terminologia de publicação de modelos, e como as capacidades de IA generativa do GKE podem melhorar e apoiar o desempenho da publicação de modelos, consulte o artigo Acerca da inferência de modelos no GKE.
vLLM
O vLLM é uma framework de serviço de LLM de código aberto altamente otimizada que aumenta a capacidade de processamento de serviço em GPUs. As principais funcionalidades incluem o seguinte:
- Implementação do transformador otimizada com PagedAttention
- Processamento em lote contínuo que melhora o débito geral da publicação
- Paralelismo de tensores e serviço distribuído em várias GPUs
Para saber mais, consulte a documentação do vLLM.
GKE Inference Gateway
O GKE Inference Gateway melhora as capacidades do GKE para publicar LLMs. Otimiza as cargas de trabalho de inferência com funcionalidades como as seguintes:
- Balanceamento de carga otimizado para inferência com base nas métricas de carga.
- Suporte para publicação densa de várias cargas de trabalho de adaptadores LoRA.
- Encaminhamento com reconhecimento do modelo para operações simplificadas.
Para mais informações, consulte o artigo Acerca do GKE Inference Gateway.
Objetivos
Antes de começar
- Sign in to your Google Cloud account. If you're new to Google Cloud, create an account to evaluate how our products perform in real-world scenarios. New customers also get $300 in free credits to run, test, and deploy workloads.
-
In the Google Cloud console, on the project selector page, select or create a Google Cloud project.
Roles required to select or create a project
- Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
-
Create a project: To create a project, you need the Project Creator
(
roles/resourcemanager.projectCreator
), which contains theresourcemanager.projects.create
permission. Learn how to grant roles.
-
Verify that billing is enabled for your Google Cloud project.
-
Enable the required API.
Roles required to enable APIs
To enable APIs, you need the Service Usage Admin IAM role (
roles/serviceusage.serviceUsageAdmin
), which contains theserviceusage.services.enable
permission. Learn how to grant roles. -
In the Google Cloud console, on the project selector page, select or create a Google Cloud project.
Roles required to select or create a project
- Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
-
Create a project: To create a project, you need the Project Creator
(
roles/resourcemanager.projectCreator
), which contains theresourcemanager.projects.create
permission. Learn how to grant roles.
-
Verify that billing is enabled for your Google Cloud project.
-
Enable the required API.
Roles required to enable APIs
To enable APIs, you need the Service Usage Admin IAM role (
roles/serviceusage.serviceUsageAdmin
), which contains theserviceusage.services.enable
permission. Learn how to grant roles. -
Make sure that you have the following role or roles on the project: roles/container.admin, roles/iam.serviceAccountAdmin
Check for the roles
-
In the Google Cloud console, go to the IAM page.
Go to IAM - Select the project.
-
In the Principal column, find all rows that identify you or a group that you're included in. To learn which groups you're included in, contact your administrator.
- For all rows that specify or include you, check the Role column to see whether the list of roles includes the required roles.
Grant the roles
-
In the Google Cloud console, go to the IAM page.
Aceder ao IAM - Selecione o projeto.
- Clique em Conceder acesso.
-
No campo Novos responsáveis, introduza o identificador do utilizador. Normalmente, este é o endereço de email de uma Conta Google.
- Na lista Selecionar uma função, selecione uma função.
- Para conceder funções adicionais, clique em Adicionar outra função e adicione cada função adicional.
- Clique em Guardar.
- Crie uma conta Hugging Face, se ainda não tiver uma.
- Certifique-se de que o seu projeto tem quota suficiente para GPUs H100. Para saber mais, consulte os artigos Planeie a quota de GPUs e Quotas de atribuição.
- Aceda à página de consentimento e valide o consentimento para utilizar a sua conta do Hugging Face.
- Aceite os termos do modelo.
- Clique em O seu perfil > Definições > Tokens de acesso.
- Selecione Novo token.
- Especifique um nome à sua escolha e uma função de, pelo menos,
Read
. - Selecione Gerar um token.
- Copie o token gerado para a área de transferência.
Na Google Cloud consola, inicie uma sessão do Cloud Shell clicando em
Ativar Cloud Shell na Google Cloud consola. Esta ação inicia uma sessão no painel inferior da consola. Google Cloud
Defina as variáveis de ambiente predefinidas:
gcloud config set project PROJECT_ID gcloud config set billing/quota_project PROJECT_ID export PROJECT_ID=$(gcloud config get project) export REGION=REGION export CLUSTER_NAME=CLUSTER_NAME export HF_TOKEN=HF_TOKEN
Substitua os seguintes valores:
PROJECT_ID
: o seu Google Cloud ID do projeto.REGION
: uma região que suporta o tipo de acelerador que quer usar, por exemplo,us-central1
para a GPU H100.CLUSTER_NAME
: o nome do cluster.HF_TOKEN
: o token do Hugging Face que gerou anteriormente.
PROJECT_ID
: o seu Google Cloud ID do projeto.CONTROL_PLANE_LOCATION
: a região do Compute Engine do plano de controlo do seu cluster. Indique uma região que suporte o tipo de acelerador que quer usar, por exemplo,us-central1
para a GPU H100.CLUSTER_NAME
: o nome do cluster.No Cloud Shell, execute o seguinte comando para criar um cluster Standard:
gcloud container clusters create CLUSTER_NAME \ --project=PROJECT_ID \ --location=CONTROL_PLANE_LOCATION \ --workload-pool=PROJECT_ID.svc.id.goog \ --release-channel=rapid \ --num-nodes=1 \ --enable-managed-prometheus \ --monitoring=SYSTEM,DCGM \ --gateway-api=standard
Substitua os seguintes valores:
PROJECT_ID
: o seu Google Cloud ID do projeto.CONTROL_PLANE_LOCATION
: a região do Compute Engine do plano de controlo do seu cluster. Indique uma região que suporte o tipo de acelerador que quer usar, por exemplo,us-central1
para a GPU H100.CLUSTER_NAME
: o nome do cluster.
A criação do cluster pode demorar vários minutos.
Para criar um node pool com o tamanho do disco adequado para executar o modelo
Llama-3.1-8B-Instruct
, execute o seguinte comando:gcloud container node-pools create gpupool \ --accelerator type=nvidia-h100-80gb,count=2,gpu-driver-version=latest \ --project=PROJECT_ID \ --location=CONTROL_PLANE_LOCATION \ --node-locations=CONTROL_PLANE_LOCATION-a \ --cluster=CLUSTER_NAME \ --machine-type=a3-highgpu-2g \ --num-nodes=1 \
O GKE cria um único node pool que contém uma GPU H100.
Para comunicar com o cluster, configure o seguinte:
kubectl
gcloud container clusters get-credentials CLUSTER_NAME \ --location=CONTROL_PLANE_LOCATION
Substitua os seguintes valores:
CONTROL_PLANE_LOCATION
: a região do Compute Engine do plano de controlo do seu cluster.CLUSTER_NAME
: o nome do cluster.
Crie um segredo do Kubernetes que contenha o token do Hugging Face:
kubectl create secret generic hf-token \ --from-literal=token=HF_TOKEN \ --dry-run=client -o yaml | kubectl apply -f -
Substitua
HF_TOKEN
pelo token do Hugging Face que gerou anteriormente.Para implementar num tipo de acelerador
nvidia-h100-80gb
, guarde o seguinte manifesto comovllm-llama3.1-8b-instruct.yaml
. Este manifesto define uma implementação do Kubernetes com o seu modelo e servidor de modelos:apiVersion: apps/v1 kind: Deployment metadata: name: vllm-llama3.1-8b-instruct spec: replicas: 3 selector: matchLabels: app: vllm-llama3.1-8b-instruct template: metadata: labels: app: vllm-llama3.1-8b-instruct spec: containers: - name: vllm image: "vllm/vllm-openai:latest" imagePullPolicy: Always command: ["python3", "-m", "vllm.entrypoints.openai.api_server"] args: - "--model" - "meta-llama/Llama-3.1-8B-Instruct" - "--tensor-parallel-size" - "1" - "--port" - "8000" - "--enable-lora" - "--max-loras" - "2" - "--max-cpu-loras" - "12" env: # Enabling LoRA support temporarily disables automatic v1, we want to force it on # until 0.8.3 vLLM is released. - name: VLLM_USE_V1 value: "1" - name: PORT value: "8000" - name: HUGGING_FACE_HUB_TOKEN valueFrom: secretKeyRef: name: hf-token key: token - name: VLLM_ALLOW_RUNTIME_LORA_UPDATING value: "true" ports: - containerPort: 8000 name: http protocol: TCP lifecycle: preStop: # vLLM stops accepting connections when it receives SIGTERM, so we need to sleep # to give upstream gateways a chance to take us out of rotation. The time we wait # is dependent on the time it takes for all upstreams to completely remove us from # rotation. Older or simpler load balancers might take upwards of 30s, but we expect # our deployment to run behind a modern gateway like Envoy which is designed to # probe for readiness aggressively. sleep: # Upstream gateway probers for health should be set on a low period, such as 5s, # and the shorter we can tighten that bound the faster that we release # accelerators during controlled shutdowns. However, we should expect variance, # as load balancers may have internal delays, and we don't want to drop requests # normally, so we're often aiming to set this value to a p99 propagation latency # of readiness -> load balancer taking backend out of rotation, not the average. # # This value is generally stable and must often be experimentally determined on # for a given load balancer and health check period. We set the value here to # the highest value we observe on a supported load balancer, and we recommend # tuning this value down and verifying no requests are dropped. # # If this value is updated, be sure to update terminationGracePeriodSeconds. # seconds: 30 # # IMPORTANT: preStop.sleep is beta as of Kubernetes 1.30 - for older versions # replace with this exec action. #exec: # command: # - /usr/bin/sleep # - 30 livenessProbe: httpGet: path: /health port: http scheme: HTTP # vLLM's health check is simple, so we can more aggressively probe it. Liveness # check endpoints should always be suitable for aggressive probing. periodSeconds: 1 successThreshold: 1 # vLLM has a very simple health implementation, which means that any failure is # likely significant. However, any liveness triggered restart requires the very # large core model to be reloaded, and so we should bias towards ensuring the # server is definitely unhealthy vs immediately restarting. Use 5 attempts as # evidence of a serious problem. failureThreshold: 5 timeoutSeconds: 1 readinessProbe: httpGet: path: /health port: http scheme: HTTP # vLLM's health check is simple, so we can more aggressively probe it. Readiness # check endpoints should always be suitable for aggressive probing, but may be # slightly more expensive than readiness probes. periodSeconds: 1 successThreshold: 1 # vLLM has a very simple health implementation, which means that any failure is # likely significant, failureThreshold: 1 timeoutSeconds: 1 # We set a startup probe so that we don't begin directing traffic or checking # liveness to this instance until the model is loaded. startupProbe: # Failure threshold is when we believe startup will not happen at all, and is set # to the maximum possible time we believe loading a model will take. In our # default configuration we are downloading a model from HuggingFace, which may # take a long time, then the model must load into the accelerator. We choose # 10 minutes as a reasonable maximum startup time before giving up and attempting # to restart the pod. # # IMPORTANT: If the core model takes more than 10 minutes to load, pods will crash # loop forever. Be sure to set this appropriately. failureThreshold: 600 # Set delay to start low so that if the base model changes to something smaller # or an optimization is deployed, we don't wait unnecessarily. initialDelaySeconds: 2 # As a startup probe, this stops running and so we can more aggressively probe # even a moderately complex startup - this is a very important workload. periodSeconds: 1 httpGet: # vLLM does not start the OpenAI server (and hence make /health available) # until models are loaded. This may not be true for all model servers. path: /health port: http scheme: HTTP resources: limits: nvidia.com/gpu: 1 requests: nvidia.com/gpu: 1 volumeMounts: - mountPath: /data name: data - mountPath: /dev/shm name: shm - name: adapters mountPath: "/adapters" # This is the second container in the Pod, a sidecar to the vLLM container. # It watches the ConfigMap and downloads LoRA adapters. - name: lora-adapter-syncer image: us-central1-docker.pkg.dev/k8s-staging-images/gateway-api-inference-extension/lora-syncer:main imagePullPolicy: Always env: - name: DYNAMIC_LORA_ROLLOUT_CONFIG value: "/config/configmap.yaml" volumeMounts: # DO NOT USE subPath, dynamic configmap updates don't work on subPaths - name: config-volume mountPath: /config restartPolicy: Always # vLLM allows VLLM_PORT to be specified as an environment variable, but a user might # create a 'vllm' service in their namespace. That auto-injects VLLM_PORT in docker # compatible form as `tcp://<IP>:<PORT>` instead of the numeric value vLLM accepts # causing CrashLoopBackoff. Set service environment injection off by default. enableServiceLinks: false # Generally, the termination grace period needs to last longer than the slowest request # we expect to serve plus any extra time spent waiting for load balancers to take the # model server out of rotation. # # An easy starting point is the p99 or max request latency measured for your workload, # although LLM request latencies vary significantly if clients send longer inputs or # trigger longer outputs. Since steady state p99 will be higher than the latency # to drain a server, you may wish to slightly this value either experimentally or # via the calculation below. # # For most models you can derive an upper bound for the maximum drain latency as # follows: # # 1. Identify the maximum context length the model was trained on, or the maximum # allowed length of output tokens configured on vLLM (llama2-7b was trained to # 4k context length, while llama3-8b was trained to 128k). # 2. Output tokens are the more compute intensive to calculate and the accelerator # will have a maximum concurrency (batch size) - the time per output token at # maximum batch with no prompt tokens being processed is the slowest an output # token can be generated (for this model it would be about 10ms TPOT at a max # batch size around 50, or 100 tokens/sec) # 3. Calculate the worst case request duration if a request starts immediately # before the server stops accepting new connections - generally when it receives # SIGTERM (for this model that is about 4096 / 100 ~ 40s) # 4. If there are any requests generating prompt tokens that will delay when those # output tokens start, and prompt token generation is roughly 6x faster than # compute-bound output token generation, so add 40% to the time from above (40s + # 16s = 56s) # # Thus we think it will take us at worst about 56s to complete the longest possible # request the model is likely to receive at maximum concurrency (highest latency) # once requests stop being sent. # # NOTE: This number will be lower than steady state p99 latency since we stop receiving # new requests which require continuous prompt token computation. # NOTE: The max timeout for backend connections from gateway to model servers should # be configured based on steady state p99 latency, not drain p99 latency # # 5. Add the time the pod takes in its preStop hook to allow the load balancers to # stop sending us new requests (56s + 30s = 86s). # # Because the termination grace period controls when the Kubelet forcibly terminates a # stuck or hung process (a possibility due to a GPU crash), there is operational safety # in keeping the value roughly proportional to the time to finish serving. There is also # value in adding a bit of extra time to deal with unexpectedly long workloads. # # 6. Add a 50% safety buffer to this time (86s * 1.5 ≈ 130s). # # One additional source of drain latency is that some workloads may run close to # saturation and have queued requests on each server. Since traffic in excess of the # max sustainable QPS will result in timeouts as the queues grow, we assume that failure # to drain in time due to excess queues at the time of shutdown is an expected failure # mode of server overload. If your workload occasionally experiences high queue depths # due to periodic traffic, consider increasing the safety margin above to account for # time to drain queued requests. terminationGracePeriodSeconds: 130 nodeSelector: cloud.google.com/gke-accelerator: "nvidia-h100-80gb" volumes: - name: data emptyDir: {} - name: shm emptyDir: medium: Memory - name: adapters emptyDir: {} - name: config-volume configMap: name: vllm-llama3.1-8b-adapters --- apiVersion: v1 kind: ConfigMap metadata: name: vllm-llama3.1-8b-adapters data: configmap.yaml: | vLLMLoRAConfig: name: vllm-llama3.1-8b-instruct port: 8000 defaultBaseModel: meta-llama/Llama-3.1-8B-Instruct ensureExist: models: - id: food-review source: Kawon/llama3.1-food-finetune_v14_r8 - id: cad-fabricator source: redcathode/fabricator --- kind: HealthCheckPolicy apiVersion: networking.gke.io/v1 metadata: name: health-check-policy namespace: default spec: targetRef: group: "inference.networking.k8s.io" kind: InferencePool name: vllm-llama3.1-8b-instruct default: config: type: HTTP httpHealthCheck: requestPath: /health port: 8000
Aplique o manifesto ao cluster:
kubectl apply -f vllm-llama3.1-8b-instruct.yaml
selector
: especifica a que agrupamentos pertence este conjunto. As etiquetas neste seletor têm de corresponder exatamente às etiquetas aplicadas aos pods do servidor do modelo.targetPort
: define as portas usadas pelo servidor de modelos nos agrupamentos.inferencePool.modelServers.matchLabels.app
: a chave da etiqueta usada para selecionar os pods do servidor do modelo.Guarde o seguinte manifesto de exemplo como
inferenceobjective.yaml
:apiVersion: inference.networking.x-k8s.io/v1alpha2 kind: InferenceObjective metadata: name: MODEL_NAME spec: priority: VALUE poolRef: name: INFERENCE_POOL_NAME kind: "InferencePool"
Substitua o seguinte:
MODEL_NAME
: o nome do modelo base ou do adaptador LoRA. Por exemplo,food-review
.VALUE
: a prioridade do objetivo de inferência. Este é um número inteiro em que um valor mais elevado indica um pedido mais crítico. Por exemplo,10
.INFERENCE_POOL_NAME
: o nome doInferencePool
que criou no passo anterior. Por exemplo,vllm-llama3.1-8b-instruct
.
Aplique o manifesto de exemplo ao seu cluster:
kubectl apply -f inferenceobjective.yaml
Guarde o seguinte manifesto de exemplo como
gateway.yaml
:apiVersion: gateway.networking.k8s.io/v1 kind: Gateway metadata: name: GATEWAY_NAME spec: gatewayClassName: gke-l7-regional-external-managed listeners: - protocol: HTTP # Or HTTPS for production port: 80 # Or 443 for HTTPS name: http
Substitua
GATEWAY_NAME
por um nome exclusivo para o recurso de gateway. Por exemplo,inference-gateway
.Aplique o manifesto ao cluster:
kubectl apply -f gateway.yaml
Guarde o seguinte manifesto de exemplo como
httproute.yaml
:apiVersion: gateway.networking.k8s.io/v1 kind: HTTPRoute metadata: name: HTTPROUTE_NAME spec: parentRefs: - name: GATEWAY_NAME rules: - matches: - path: type: PathPrefix value: PATH_PREFIX backendRefs: - name: INFERENCE_POOL_NAME group: inference.networking.k8s.io kind: InferencePool
Substitua o seguinte:
HTTPROUTE_NAME
: um nome exclusivo para o seu recursoHTTPRoute
. Por exemplo,my-route
.GATEWAY_NAME
: o nome do recursoGateway
que criou. Por exemplo,inference-gateway
.PATH_PREFIX
: o prefixo do caminho que usa para fazer corresponder pedidos recebidos. Por exemplo,/
para corresponder a tudo.INFERENCE_POOL_NAME
: o nome do recursoInferencePool
para o qual quer encaminhar o tráfego. Por exemplo,vllm-llama3.1-8b-instruct
.
Aplique o manifesto ao cluster:
kubectl apply -f httproute.yaml
- Obtenha o ponto final do gateway.
- Construa um pedido JSON formatado corretamente.
- Use
curl
para enviar o pedido para o ponto final/v1/completions
. Para obter o ponto final do gateway, execute o seguinte comando:
IP=$(kubectl get gateway/GATEWAY_NAME -o jsonpath='{.status.addresses[0].value}') PORT=80
Substitua
GATEWAY_NAME
pelo nome do recurso de gateway.Para enviar um pedido para o ponto final
/v1/completions
através docurl
, execute o seguinte comando:curl -i -X POST http://${IP}:${PORT}/v1/completions \ -H "Content-Type: application/json" \ -d '{ "model": "MODEL_NAME", "prompt": "PROMPT_TEXT", "max_tokens": MAX_TOKENS, "temperature": "TEMPERATURE" }'
Substitua o seguinte:
MODEL_NAME
: o nome do modelo ou do adaptador LoRA a usar.PROMPT_TEXT
: o comando de entrada para o modelo.MAX_TOKENS
: o número máximo de tokens a gerar na resposta.TEMPERATURE
: controla a aleatoriedade do resultado. Use o valor0
para uma saída determinística ou um número superior para uma saída mais criativa.
- Corpo do pedido: o corpo do pedido pode incluir parâmetros adicionais, como
stop
etop_p
. Consulte a especificação da API OpenAI para ver uma lista completa de opções. - Processamento de erros: implemente o processamento de erros adequado no código do cliente para processar potenciais erros na resposta. Por exemplo, verifique o código de estado HTTP na resposta
curl
. Geralmente, um código de estado diferente de 200 indica um erro. - Autenticação e autorização: para implementações de produção, proteja o ponto final da API com mecanismos de autenticação e autorização. Inclua os cabeçalhos
adequados (por exemplo,
Authorization
) nos seus pedidos. CONTROL_PLANE_LOCATION
: a região do Compute Engine do plano de controlo do seu cluster.CLUSTER_NAME
: o nome do cluster.- Leia sobre o GKE Inference Gateway.
- Leia sobre a implementação do GKE Inference Gateway.
Aceda ao modelo
Para implementar o modelo
Llama3.1
no GKE, assine o contrato de consentimento de licença e gere uma chave de acesso da Hugging Face.Assine o contrato de consentimento de licença
Tem de assinar o contrato de consentimento para usar o modelo
Llama3.1
. Siga estas instruções:Gere um token de acesso
Para aceder ao modelo através do Hugging Face, precisa de um token do Hugging Face.
Siga estes passos para gerar um novo token se ainda não tiver um:
Prepare o seu ambiente
Neste tutorial, vai usar o Cloud Shell para gerir recursos alojados no Google Cloud. O Cloud Shell é pré-instalado com o software necessário para este tutorial, incluindo
kubectl
e a CLI gcloud.Para configurar o seu ambiente com o Cloud Shell, siga estes passos:
Crie e configure Google Cloud recursos
Crie um cluster e um node pool do GKE
Publicar MDIs em GPUs num cluster do GKE Autopilot ou Standard. Recomendamos que use um cluster do Autopilot para uma experiência do Kubernetes totalmente gerida. Para escolher o modo de funcionamento do GKE mais adequado às suas cargas de trabalho, consulte o artigo Escolha um modo de funcionamento do GKE.
Piloto automático
No Cloud Shell, execute o seguinte comando:
gcloud container clusters create-auto CLUSTER_NAME \ --project=PROJECT_ID \ --location=CONTROL_PLANE_LOCATION \ --release-channel=rapid
Substitua os seguintes valores:
O GKE cria um cluster do Autopilot com nós de CPU e GPU, conforme solicitado pelas cargas de trabalho implementadas.
Standard
Configure a autorização para extrair métricas
Para configurar a autorização para extrair métricas, crie o segredo
inference-gateway-sa-metrics-reader-secret
:kubectl apply -f - <<EOF --- apiVersion: rbac.authorization.k8s.io/v1 kind: ClusterRole metadata: name: inference-gateway-metrics-reader rules: - nonResourceURLs: - /metrics verbs: - get --- apiVersion: v1 kind: ServiceAccount metadata: name: inference-gateway-sa-metrics-reader namespace: default --- apiVersion: rbac.authorization.k8s.io/v1 kind: ClusterRoleBinding metadata: name: inference-gateway-sa-metrics-reader-role-binding namespace: default subjects: - kind: ServiceAccount name: inference-gateway-sa-metrics-reader namespace: default roleRef: kind: ClusterRole name: inference-gateway-metrics-reader apiGroup: rbac.authorization.k8s.io --- apiVersion: v1 kind: Secret metadata: name: inference-gateway-sa-metrics-reader-secret namespace: default annotations: kubernetes.io/service-account.name: inference-gateway-sa-metrics-reader type: kubernetes.io/service-account-token --- apiVersion: rbac.authorization.k8s.io/v1 kind: ClusterRole metadata: name: inference-gateway-sa-metrics-reader-secret-read rules: - resources: - secrets apiGroups: [""] verbs: ["get", "list", "watch"] resourceNames: ["inference-gateway-sa-metrics-reader-secret"] --- apiVersion: rbac.authorization.k8s.io/v1 kind: ClusterRoleBinding metadata: name: gmp-system:collector:inference-gateway-sa-metrics-reader-secret-read namespace: default roleRef: name: inference-gateway-sa-metrics-reader-secret-read kind: ClusterRole apiGroup: rbac.authorization.k8s.io subjects: - name: collector namespace: gmp-system kind: ServiceAccount EOF
Crie um segredo do Kubernetes para as credenciais do Hugging Face
No Cloud Shell, faça o seguinte:
Instale os CRDs
InferenceObjective
eInferencePool
Nesta secção, instala as definições de recursos personalizados (CRDs) necessárias para o GKE Inference Gateway.
Os CRDs expandem a API Kubernetes. Isto permite-lhe definir novos tipos de recursos. Para usar o GKE Inference Gateway, instale os CRDs
InferencePool
eInferenceObjective
no seu cluster do GKE executando o seguinte comando:kubectl apply -f https://github.com/kubernetes-sigs/gateway-api-inference-extension/releases/download/v1.0.0/manifests.yaml
Implemente o servidor de modelos
Este exemplo implementa um modelo
Llama3.1
através de um servidor de modelos vLLM. A implementação está etiquetada comoapp:vllm-llama3.1-8b-instruct
. Esta implementação também usa dois adaptadores LoRA denominadosfood-review
ecad-fabricator
da Hugging Face. Pode atualizar esta implementação com o seu próprio servidor de modelos e contentor de modelos, porta de publicação e nome da implementação. Opcionalmente, pode configurar adaptadores LoRA na implementação ou implementar o modelo base.Crie um recurso
InferencePool
O recurso personalizado do Kubernetes
InferencePool
define um grupo de pods com uma configuração de computação e um MDG base comuns.O recurso personalizado
InferencePool
inclui os seguintes campos principais:O recurso
InferencePool
permite que o GKE Inference Gateway encaminhe o tráfego para os pods do servidor de modelos.Para criar um
InferencePool
com o Helm, siga estes passos:helm install vllm-llama3.1-8b-instruct \ --set inferencePool.modelServers.matchLabels.app=vllm-llama3.1-8b-instruct \ --set provider.name=gke \ --set healthCheckPolicy.create=false \ --version v1.0.0 \ oci://registry.k8s.io/gateway-api-inference-extension/charts/inferencepool
Altere o campo seguinte para corresponder à sua implementação:
Este comando cria um objeto
InferencePool
que representa logicamente a implementação do servidor de modelos e faz referência aos serviços de ponto final do modelo nos pods que oSelector
seleciona.Crie um recurso
InferenceObjective
com uma criticidade de publicaçãoO recurso personalizado
InferenceObjective
define os parâmetros de publicação de um modelo, incluindo a respetiva prioridade. Tem de criar recursosInferenceObjective
para definir que modelos são publicados numInferencePool
. Estes recursos podem fazer referência a modelos base ou adaptadores LoRA suportados pelos servidores de modelos noInferencePool
.O campo
metadata.name
especifica o nome do modelo, o campopriority
define a respetiva criticidade de publicação e o campopoolRef
estabelece uma ligação aoInferencePool
onde o modelo é publicado.Para criar um
InferenceObjective
, siga estes passos:O exemplo seguinte cria dois objetos
InferenceObjective
. A primeira configura ofood-review
modelo LoRA novllm-llama3.1-8b-instruct
InferencePool
com uma prioridade de10
. A segunda configuração define que ollama3-base-model
seja publicado com uma prioridade mais elevada de20
.apiVersion: inference.networking.k8s.io/v1alpha1 kind: InferenceObjective metadata: name: food-review spec: priority: 10 poolRef: name: vllm-llama3.1-8b-instruct kind: "InferencePool" --- apiVersion: inference.networking.k8s.io/v1alpha1 kind: InferenceObjective metadata: name: llama3-base-model spec: priority: 20 poolRef: name: vllm-llama3.1-8b-instruct kind: "InferencePool"
Crie o gateway
O recurso Gateway funciona como o ponto de entrada para o tráfego externo no seu cluster do Kubernetes. Define os ouvintes que aceitam ligações recebidas.
O GKE Inference Gateway suporta a classe de gateway
gke-l7-rilb
egke-l7-regional-external-managed
. Para mais informações, consulte a documentação do GKE sobre classes de gateway.Para criar um gateway, siga estes passos:
Crie o recurso
HTTPRoute
Nesta secção, cria um recurso
HTTPRoute
para definir como o gateway encaminha os pedidos HTTP recebidos para o seuInferencePool
.O recurso HTTPRoute define como o GKE Gateway encaminha os pedidos HTTP recebidos para serviços de back-end, que é o seu
InferencePool
. Especifica regras de correspondência (por exemplo, cabeçalhos ou caminhos) e o back-end para o qual o tráfego deve ser encaminhado.Para criar um HTTPRoute, siga estes passos:
Envie um pedido de inferência
Depois de configurar o GKE Inference Gateway, pode enviar pedidos de inferência para o modelo implementado.
Para enviar pedidos de inferência, siga estes passos:
Isto permite-lhe gerar texto com base no seu comando de entrada e nos parâmetros especificados.
Tenha em atenção o seguinte:
Configure a observabilidade para o seu Inference Gateway
O GKE Inference Gateway oferece observabilidade sobre o estado, o desempenho e o comportamento das suas cargas de trabalho de inferência. Isto ajuda a identificar e resolver problemas, otimizar a utilização de recursos e garantir a fiabilidade das suas aplicações. Pode ver estas métricas de observabilidade no Cloud Monitoring através do Explorador de métricas.
Para configurar a observabilidade para o GKE Inference Gateway, consulte o artigo Configure a observabilidade.
Elimine os recursos implementados
Para evitar incorrer em custos na sua Google Cloud conta pelos recursos que criou a partir deste guia, execute o seguinte comando:
gcloud container clusters delete CLUSTER_NAME \ --location=CONTROL_PLANE_LOCATION
Substitua os seguintes valores:
O que se segue?
-