View a markdown version of this page

Implemente modelos de Amazon S3, Amazon FSx o Hugging Face Hub mediante kubectl - Amazon SageMaker AI
Requisitos previosAjustes y configuraciónImplemente su modelo desde Amazon S3, Amazon FSx o Hugging Face HubVerificación del estado de la implementaciónAdministración de la implementación

Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.

Implemente modelos de Amazon S3, Amazon FSx o Hugging Face Hub mediante kubectl

Los siguientes pasos muestran cómo implementar modelos almacenados en Amazon S3, Amazon FSx o Hugging Face Hub en un clúster de SageMaker HyperPod Amazon mediante kubectl.

Las siguientes instrucciones contienen celdas de código y comandos diseñados para ejecutarse en una terminal. Asegúrese de haber configurado su entorno con AWS credenciales antes de ejecutar estos comandos.

Requisitos previos

Antes de comenzar, compruebe que:

Ajustes y configuración

Reemplace todos los valores marcadores de posición por los identificadores de recursos reales.

  1. Seleccione su región en su entorno.

    export REGION=<region>

  2. Inicialice el nombre de su clúster. Esto identifica el HyperPod clúster en el que se implementará el modelo.

    nota

    Consulte con el administrador del clúster para asegurarse de que se concedan los permisos a ese rol o usuario. Puede ejecutar !aws sts get-caller-identity --query "Arn" para comprobar qué rol o usuario se está utilizando en la terminal.

    # Specify your hyperpod cluster name here
HYPERPOD_CLUSTER_NAME="<Hyperpod_cluster_name>"

# NOTE: For sample deployment, we use g5.24xlarge for Llama 3.1 8B model which has sufficient memory and GPU
instance_type="ml.g5.24xlarge"

  3. Inicialice el espacio de nombres de su clúster. El administrador del clúster ya debería haber creado una cuenta de servicio de inferencia de hiperpods en su espacio de nombres.

    cluster_namespace="<namespace>"

  4. Cree una CRD con una de las siguientes opciones:

    Using Amazon FSx as the model source

    1. Configure un nombre de SageMaker punto final.

      export SAGEMAKER_ENDPOINT_NAME="llama-fsx"

    2. Configure el ID del sistema de archivos de Amazon FSx que se va a utilizar.

      export FSX_FILE_SYSTEM_ID="fs-1234abcd"

    3. A continuación, se muestra un ejemplo de archivo yaml para crear un punto final con Amazon FSx y un modelo Llama.

      nota

      Para los clústeres con la partición de GPU habilitada, sustitúyala por nvidia.com/gpu el nombre de recurso MIG correspondiente, como. nvidia.com/mig-1g.10gb Para obtener más información, consulte Envío de tareas con MIG.

      cat <<EOF> deploy_fsx_cluster_inference.yaml
---
apiVersion: inference.sagemaker.aws.amazon.com/v1
kind: InferenceEndpointConfig
metadata:
  name: $SAGEMAKER_ENDPOINT_NAME
  namespace: $CLUSTER_NAMESPACE
spec:
  modelName: Llama-3.1-8B-Instruct
  instanceType: ml.g5.24xlarge
  invocationEndpoint: v1/chat/completions
  replicas: 2
  modelSourceConfig:
    fsxStorage:
      fileSystemId: $FSX_FILE_SYSTEM_ID
    modelLocation: Llama-3.1-8B-Instruct
    modelSourceType: fsx
  worker:
    image: vllm/vllm-openai:v0.19.1
    modelInvocationPort:
      containerPort: 8000
      name: http
    modelVolumeMount:
      mountPath: /opt/ml/model
      name: model-weights
    resources:
      limits:
        nvidia.com/gpu: 4
      requests:
        cpu: 30000m
        memory: 100Gi
        nvidia.com/gpu: 4
    args:
      - "--model"
      - "/opt/ml/model"
      - "--port"
      - "8000"
      - "--tensor-parallel-size"
      - "4"
      - "--served-model-name"
      - "Llama-3.1-8B-Instruct"
    environmentVariables:
      - name: VLLM_REQUEST_TIMEOUT
        value: "600"
EOF
    Using Amazon S3 as the model source

    1. Configure un nombre de punto SageMaker final.

      export SAGEMAKER_ENDPOINT_NAME="llama-s3"

    2. Configure la ubicación del bucket de Amazon S3 donde se encuentra el modelo.

      export S3_MODEL_LOCATION="<your-s3-bucket-name>"

    3. A continuación, se muestra un ejemplo de archivo yaml para crear un punto final con Amazon S3 y un modelo Llama que utilice vLLM como tiempo de ejecución de inferencias.

      nota

      En el caso de clústeres con la partición de GPU habilitada, sustitúyala por el nvidia.com/gpu nombre de recurso MIG correspondiente, como. nvidia.com/mig-1g.10gb Para obtener más información, consulte Envío de tareas con MIG.

      cat <<EOF> deploy_s3_inference.yaml
---
apiVersion: inference.sagemaker.aws.amazon.com/v1
kind: InferenceEndpointConfig
metadata:
  name: $SAGEMAKER_ENDPOINT_NAME
  namespace: $CLUSTER_NAMESPACE
spec:
  modelName: Llama-3.1-8B-Instruct
  instanceType: ml.g5.24xlarge
  invocationEndpoint: v1/chat/completions
  replicas: 2
  modelSourceConfig:
    modelSourceType: s3
    s3Storage:
      bucketName: $S3_MODEL_LOCATION
      region: $REGION
    modelLocation: Llama-3.1-8B-Instruct
    prefetchEnabled: true
  worker:
    image: vllm/vllm-openai:v0.19.1
    modelInvocationPort:
      containerPort: 8000
      name: http
    modelVolumeMount:
      name: model-weights
      mountPath: /opt/ml/model
    resources:
      limits:
        nvidia.com/gpu: 4
      requests:
        cpu: 30000m
        memory: 100Gi
        nvidia.com/gpu: 4
    args:
      - "--model"
      - "/opt/ml/model"
      - "--port"
      - "8000"
      - "--tensor-parallel-size"
      - "4"
      - "--served-model-name"
      - "Llama-3.1-8B-Instruct"
    environmentVariables:
      - name: VLLM_REQUEST_TIMEOUT
        value: "600"
EOF
    Using Hugging Face Hub as the model source

    1. Crea un secreto de Kubernetes que contenga tu token de la API Hugging Face. Este token es obligatorio para los modelos cerrados y se recomienda para todas las descargas. Puedes generar un token en huggingface. co/settings/tokens.

      importante

      La implementación de modelos desde Hugging Face Hub requiere acceso saliente a Internet desde los nodos del clúster a los dominios de Hugging Face, incluidos y. *.huggingface.co *.hf.co Asegúrese de que la configuración de red de la VPC (puerta de enlace NAT, grupos de seguridad y ACL de red) permita la salida de HTTPS a estos dominios. Sin acceso a Internet, la descarga del modelo fallará.

      nota

      Para los entornos de producción, recomendamos utilizar Amazon S3 o Amazon FSx como fuente del modelo en lugar de Hugging Face Hub. Con Amazon S3 y Amazon FSx, los artefactos del modelo se almacenan en su AWS cuenta, lo que elimina la dependencia de la conectividad externa a Internet y proporciona tiempos de implementación más predecibles. Hugging Face Hub es ideal para el desarrollo, la experimentación y la creación rápida de prototipos, cuando es conveniente acceder directamente al repositorio de modelos Hugging Face.

      kubectl create secret generic hf-token-secret \
  --from-literal=token=hf_YOUR_TOKEN_HERE \
  -n $CLUSTER_NAMESPACE

    2. Configure un nombre de punto final SageMaker .

      export SAGEMAKER_ENDPOINT_NAME="mistral7b-hf"

    3. El siguiente es un ejemplo de archivo YAML para implementar un modelo Mistral 7B de Hugging Face Hub utilizando vLLM como tiempo de ejecución de inferencia. ConprefetchEnabled: true, el operador utiliza un contenedor de inicio para descargar el modelo antes de que se inicie el contenedor de inferencias.

      nota

      Para los clústeres con la partición de GPU habilitada, sustitúyala por nvidia.com/gpu el nombre de recurso MIG correspondiente, como. nvidia.com/mig-1g.10gb Para obtener más información, consulte Envío de tareas con MIG.

      cat <<EOF> deploy_hf_inference.yaml
---
apiVersion: inference.sagemaker.aws.amazon.com/v1
kind: InferenceEndpointConfig
metadata:
  name: $SAGEMAKER_ENDPOINT_NAME
  namespace: $CLUSTER_NAMESPACE
spec:
  modelName: mistral-7b
  modelSourceConfig:
    modelSourceType: huggingface
    prefetchEnabled: true
    huggingFaceModel:
      modelId: "mistralai/Mistral-7B-Instruct-v0.3"
      tokenSecretRef:
        name: hf-token-secret
        key: token
  instanceType: "ml.g5.24xlarge"
  invocationEndpoint: v1/chat/completions
  worker:
    image: "vllm/vllm-openai:v0.19.1"
    modelInvocationPort:
      containerPort: 8000
      name: http
    modelVolumeMount:
      name: model-weights
      mountPath: /opt/ml/model
    resources:
      requests:
        nvidia.com/gpu: "4"
        memory: "96Gi"
        cpu: "16"
      limits:
        nvidia.com/gpu: "4"
        memory: "96Gi"
        cpu: "16"
    args:
      - "--model"
      - "/opt/ml/model"
      - "--port"
      - "8000"
      - "--tensor-parallel-size"
      - "4"
      - "--served-model-name"
      - "mistralai/Mistral-7B-Instruct-v0.3"
    environmentVariables:
      - name: VLLM_REQUEST_TIMEOUT
        value: "600"
EOF

    4. Los campos clave de configuración de Hugging Face son:

      • modelSourceType(obligatorio): se establece en. huggingface

      • huggingFaceModel.modelId(obligatorio): el identificador del modelo Hugging Face Hub org/model en formato (por ejemplomistralai/Mistral-7B-Instruct-v0.3,).

      • huggingFaceModel.commitSHA(opcional): un Git de 40 caracteres compromete a SHA a fijar una versión específica del modelo. Si se omite, el valor predeterminado es la rama. main

      • huggingFaceModel.tokenSecretRef(opcional): referencia a un secreto de Kubernetes que contiene tu token de la API Hugging Face. Necesario para los modelos cerrados. El token solo se usa durante la descarga del modelo y no está expuesto al contenedor de inferencias.

      • prefetchEnabled(opcional): cuando true un contenedor de inicio descarga el modelo antes de que se inicie el contenedor de inferencia. Cuandofalse, el tiempo de ejecución de la inferencia (vLLM, TGI, SGLang) descarga el modelo de forma nativa al iniciarse. El valor predeterminado es false.
nota

Para configurar el almacenamiento en caché KV y el enrutamiento inteligente para mejorar el rendimiento, consulte. Configure el almacenamiento en caché KV y el enrutamiento inteligente

Implemente su modelo desde Amazon S3, Amazon FSx o Hugging Face Hub

  1. Obtenga el nombre del clúster de Amazon EKS del ARN del HyperPod clúster para la autenticación de kubectl.

    export EKS_CLUSTER_NAME=$(aws --region $REGION sagemaker describe-cluster --cluster-name $HYPERPOD_CLUSTER_NAME \
  --query 'Orchestrator.Eks.ClusterArn' --output text | \
  cut -d'/' -f2)
aws eks update-kubeconfig --name $EKS_CLUSTER_NAME --region $REGION

  2. Implemente su InferenceEndpointConfig modelo con una de las siguientes opciones:

    Deploy with Amazon FSx as a source
    kubectl apply -f deploy_fsx_luster_inference.yaml
    Deploy with Amazon S3 as a source
    kubectl apply -f deploy_s3_inference.yaml
    Deploy with Hugging Face Hub as a source
    kubectl apply -f deploy_hf_inference.yaml

    Si la implementación falla, compruebe los InferenceEndpointConfig eventos para obtener información de diagnóstico. Para obtener información sobre problemas comunes, como errores de token, conectividad de red o falta de modelo, consulteFallos en la implementación del modelo Hugging Face Hub.

Verificación del estado de la implementación

  1. Compruebe si el modelo se ha implementado correctamente.

    kubectl describe InferenceEndpointConfig $SAGEMAKER_ENDPOINT_NAME -n $CLUSTER_NAMESPACE

  2. Compruebe que el punto de conexión se haya creado correctamente.

    kubectl describe SageMakerEndpointRegistration $SAGEMAKER_ENDPOINT_NAME -n $CLUSTER_NAMESPACE

  3. Pruebe el punto de conexión implementado para comprobar que funciona correctamente. Este paso confirma que el modelo se ha implementado correctamente y que puede procesar las solicitudes de inferencia.

    aws sagemaker-runtime invoke-endpoint \
  --endpoint-name $SAGEMAKER_ENDPOINT_NAME \
  --content-type "application/json" \
  --body '{"inputs": "What is AWS SageMaker?"}' \
  --region $REGION \
  --cli-binary-format raw-in-base64-out \
  /dev/stdout

Administración de la implementación

Cuando termine de probar la implementación, utilice los siguientes comandos para limpiar los recursos.

nota

Verifique que ya no necesita el modelo implementado ni los datos almacenados antes de continuar.

Eliminación de sus recursos

  1. Elimine la implementación de la inferencia y los recursos de Kubernetes asociados. Esto detiene los contenedores del modelo en ejecución y elimina el SageMaker punto final.

    kubectl delete inferenceendpointconfig $SAGEMAKER_ENDPOINT_NAME -n $CLUSTER_NAMESPACE

  2. Compruebe que la limpieza se ha realizado correctamente.

    # # Check that Kubernetes resources are removed
kubectl get pods,svc,deployment,InferenceEndpointConfig,sagemakerendpointregistration -n $CLUSTER_NAMESPACE
    # Verify SageMaker endpoint is deleted (should return error or empty)
aws sagemaker describe-endpoint --endpoint-name $SAGEMAKER_ENDPOINT_NAME --region $REGION
Resolución de problemas

Use estos comandos de depuración si la implementación no funciona según lo previsto.

  1. Compruebe el estado de la implementación de Kubernetes.

    kubectl describe deployment $SAGEMAKER_ENDPOINT_NAME -n $CLUSTER_NAMESPACE

  2. Compruebe el InferenceEndpointConfig estado para ver el estado de la implementación de alto nivel y cualquier problema de configuración.

    kubectl describe InferenceEndpointConfig $SAGEMAKER_ENDPOINT_NAME -n $CLUSTER_NAMESPACE

  3. Compruebe el estado de todos los objetos de Kubernetes. Obtendrá una visión completa de todos los recursos de Kubernetes relacionados en su espacio de nombres. Esto le proporciona una visión general rápida de lo que se está ejecutando y lo que podría faltar.

    kubectl get pods,svc,deployment,InferenceEndpointConfig,sagemakerendpointregistration -n $CLUSTER_NAMESPACE