Creación de un trabajo de optimización de inferencias

Cómo empezar a crear un trabajo de optimización

1. En SageMaker AI Studio, cree un trabajo de optimización mediante cualquiera de las siguientes rutas:

   • Para crear un trabajo para un JumpStart modelo, haga lo siguiente:

     1. En el menú de navegación, elija JumpStart.

     2. En la página Todos los modelos públicos, elija un proveedor de modelos y, a continuación, seleccione uno de los modelos que admitan la optimización.

     3. En la página de detalles del modelo, elija Optimizar. Este botón solo está habilitado para los modelos que admiten la optimización.

     4. En la página Crear un trabajo de optimización de inferencias, algunos JumpStart modelos requieren que firme un acuerdo de licencia de usuario final (EULA) antes de poder continuar. Si se solicita, revise los términos de la licencia en la sección Acuerdo de licencia. Si los términos son aceptables para el caso de uso, seleccione la casilla de verificación Acepto el EULA, y he leído los términos y las condiciones.

   • Para crear un trabajo para un JumpStart modelo ajustado, haga lo siguiente:

     1. En el panel de navegación, en Trabajos, elija Entrenamiento.

     2. En la página Trabajos de formación, elija el nombre de un trabajo que utilizó para ajustar un JumpStart modelo. Estos trabajos tienen el tipo JumpStartFormación en la columna Tipo de trabajo.

     3. En la página de detalles del trabajo de entrenamiento, elija Optimizar.

   • Para crear un trabajo para un modelo personalizado, haga lo siguiente:

     1. En el menú de navegación, en Trabajos, elija Optimización de inferencias.

     2. Elige Crear nuevo tarea.

     3. En la página Crear trabajo de optimización de inferencias, elija Agregar modelo.

     4. En la ventana Agregar modelo, elija Modelo personalizado.

     5. En Nombre de modelo personalizado, escriba un nombre.

     6. En URI de S3, introduzca el URI de la ubicación de Amazon S3 en la que ha almacenado los artefactos del modelo.

2. En la página Crear trabajo de optimización de inferencias, en Nombre del trabajo, puede aceptar el nombre predeterminado que la SageMaker IA asigna. O bien, para introducir un nombre de trabajo personalizado, seleccione el campo Nombre del trabajo y elija Introducir nombre del trabajo.

Establecimiento de las configuraciones de optimización

1. En Tipo de instancia de implementación, elija el tipo de instancia para el que desee optimizar el modelo.

   El tipo de instancia afecta a las técnicas de optimización que puede elegir. Para la mayoría de los tipos que utilizan hardware de GPU, las técnicas admitidas son Cuantización y Decodificación especulativa. Si eliges una instancia que usa silicio personalizado, como la instancia de AWS Inferentia ml.inf2.8xlarge, la técnica admitida es la compilación, que puedes usar para compilar el modelo para ese tipo de hardware específico.

2. Seleccione una o varias de las técnicas de optimización que Studio proporciona:

   • Si selecciona Cuantización, elija un tipo de datos para la opción Tipo de datos de precisión.

   • Si seleccionas Decodificación especulativa, elige una de las siguientes opciones:

     • Utilizar el modelo de borrador de SageMaker IA: elija utilizar el modelo de borrador que proporciona la SageMaker IA.

       nota

       Si opta por utilizar el modelo de borrador de SageMaker IA, también debe habilitar el aislamiento de la red. Studio ofrece esta opción en Seguridad.

     • Elegir modelo de JumpStart borrador: elija seleccionar un modelo del JumpStart catálogo para usarlo como modelo de borrador.

     • Elija su propio modelo de borrador: elija usar su propio modelo de borrador y proporcione el URI de S3 que lo localiza.

   • Si selecciona Carga rápida del modelo, Studio muestra la variable de OPTION_TENSOR_PARALLEL_DEGREE entorno. Utilice el campo Valor para establecer el grado de paralelismo del tensor. El valor debe dividir GPUs en partes iguales el número de instancias que haya elegido como tipo de instancia de despliegue. Por ejemplo, para fragmentar el modelo mientras usas una instancia con 8 GPUs, usa los valores 2, 4 u 8.

   • Si estableces el tipo de instancia de despliegue en una instancia de AWS Inferentia o AWS Trainium, Studio podría mostrar que la compilación es la única opción admitida. En tal caso, Studio selecciona esta opción automáticamente.

3. En Salida, introduzca el URI de una ubicación de Amazon S3. Allí, la SageMaker IA almacena los artefactos del modelo optimizado que crea tu trabajo.

4. (Opcional) Amplíe Opciones avanzadas para obtener un control más pormenorizado de la configuración, como, por ejemplo, el rol de IAM, la VPC y las variables de entorno. Para obtener más información, consulte a continuación Opciones avanzadas.

5. Cuando haya terminado de configurar el trabajo, seleccione Crear trabajo.

   Studio muestra la página de detalles del trabajo, que muestra el estado del trabajo y todos sus ajustes.

Configuración de un proyecto

1. En el código de la aplicación, importe las bibliotecas necesarias. En el ejemplo siguiente se importa el SDK para Python (Boto3). También importa las clases del SDK de Python para SageMaker IA que se utilizan para definir los modelos y trabajar con ellos:

   import boto3
   from sagemaker.serve.builder.model_builder import ModelBuilder
   from sagemaker.serve.builder.schema_builder import SchemaBuilder
   from sagemaker.session import Session
   from pathlib import Path

2. Inicializa una sesión de SageMaker IA. En el siguiente ejemplo se utiliza la Session() clase:

   sagemaker_session = Session()

Definición de un modelo

1. Cree una instancia de SchemaBuilder y proporcione ejemplos de entrada y salida. Al definir un modelo, debe proporcionar esta instancia a la clase ModelBuilder. Con ella, la SageMaker IA genera automáticamente las funciones de clasificación para serializar y deserializar la entrada y la salida.

   Para obtener más información sobre el uso de las clases SchemaBuilder y ModelBuilder, consulte Cree un modelo en Amazon SageMaker AI con ModelBuilder.

   En el ejemplo siguiente se proporcionan ejemplos de cadenas de entrada y salida para la clase SchemaBuilder:

   response = "Jupiter is the largest planet in the solar system. It is the fifth planet from the sun."
   sample_input = {
       "inputs": "What is the largest planet in the solar system?",
       "parameters": {"max_new_tokens": 128, "top_p": 0.9, "temperature": 0.6},
   }
   sample_output = [{"generated_text": response}]
   schema_builder = SchemaBuilder(sample_input, sample_output)

2. Defina su modelo según la IA. SageMaker En el ejemplo siguiente se definen los parámetros para inicializar una instancia de ModelBuilder:

   model_builder = ModelBuilder(
       model="jumpstart-model-id",
       schema_builder=schema_builder,
       sagemaker_session=sagemaker_session,
       role_arn=sagemaker_session.get_caller_identity_arn(),
   )

   En este ejemplo se utiliza un JumpStart modelo. jumpstart-model-idSustitúyalo por el identificador de un JumpStart modelo, por ejemplometa-textgeneration-llama-3-70b.

   nota

   Si quiere optimizar con una decodificación especulativa y quiere utilizar el borrador de SageMaker IA, debe habilitar el aislamiento de la red. Para habilitarlo, incluye el siguiente argumento al inicializar una ModelBuilder instancia:

   enable_network_isolation=True,

   Para obtener más información sobre el aislamiento de la red, consulteAislamiento de red.

Para optimizar con la cuantificación

1. Para ejecutar un trabajo de cuantificación, utilice el optimize() método y establezca el quantization_config argumento. El siguiente ejemplo se establece OPTION_QUANTIZE como una variable de entorno en el contenedor de optimización:

   optimized_model = model_builder.optimize(
       instance_type="instance-type",
       accept_eula=True,
       quantization_config={
           "OverrideEnvironment": {
               "OPTION_QUANTIZE": "awq",
           },
       },
       output_path="s3://output-path",
   )

   En este ejemplo, instance-type sustitúyala por una instancia de ML, comoml.p4d.24xlarge. s3://output-pathSustitúyala por la ruta a la ubicación S3 en la que se almacena el modelo optimizado que crea el trabajo.

   El optimize() método devuelve un Model objeto, que puede utilizar para implementar el modelo en un punto final.

2. Cuando finalice el trabajo, despliegue el modelo. En el siguiente ejemplo se utiliza el deploy() método:

   predictor = optimized_model.deploy(
       instance_type="instance-type", 
       accept_eula=True,
   )

   En este ejemplo, instance-type sustituya por una instancia de ML, comoml.p4d.24xlarge.

   El deploy() método devuelve un objeto predictor, que puede utilizar para enviar solicitudes de inferencia al punto final que aloja el modelo.

Para optimizarlo con decodificación especulativa utilizando el borrador del modelo de IA SageMaker

Cuando optimizas tu modelo con decodificación especulativa, puedes optar por usar un borrador de modelo proporcionado por la SageMaker IA o puedes usar el tuyo propio. En los ejemplos siguientes se utiliza el modelo preliminar de SageMaker IA.

1. Para ejecutar un trabajo de decodificación especulativa, utilice el optimize() método y establezca el speculative_decoding_config argumento. El siguiente ejemplo establece la ModelProvider clave SAGEMAKER para utilizar el borrador del modelo que proporciona la SageMaker IA.

   optimized_model = model_builder.optimize(
       instance_type="instance-type",
       accept_eula=True,
       speculative_decoding_config={
           "ModelProvider": "SAGEMAKER",
       },
   )

   En este ejemplo, instance-type sustitúyala por una instancia de aprendizaje automático, comoml.p4d.24xlarge.

   El optimize() método devuelve un Model objeto, que puede utilizar para implementar el modelo en un punto final.

2. Cuando finalice el trabajo, despliegue el modelo. En el siguiente ejemplo se utiliza el deploy() método:

   predictor = optimized_model.deploy(accept_eula=True)

   El deploy() método devuelve un objeto predictor, que puede utilizar para enviar solicitudes de inferencia al punto final que aloja el modelo.

Para optimizarlo mediante la decodificación especulativa mediante un modelo de borrador personalizado

Antes de poder proporcionar su borrador de modelo personalizado a SageMaker AI, primero debe cargar los artefactos del modelo en Amazon S3.

Los siguientes ejemplos muestran una forma posible de proporcionar un modelo de borrador personalizado. Los ejemplos descargan el borrador del modelo del Hugging Face Hub, lo suben a Amazon S3 y proporcionan el URI de S3 speculative_decoding_config al argumento.

1. Si quieres descargar un modelo del Hugging Face Hub, añade huggingface_hub la biblioteca a tu proyecto y descarga un modelo con snapshot_download() el método. En el siguiente ejemplo, se descarga un modelo a un directorio local:

   import huggingface_hub
   
   huggingface_hub.snapshot_download(
       repo_id="model-id",
       revision="main",
       local_dir=download-dir,
       token=hf-access-token,
   )

   En este ejemplo, model-id sustitúyalo por el identificador de un modelo, Hugging Face Hub, comometa-llama/Meta-Llama-3-8B. download-dirSustitúyalo por un directorio local. hf-access-tokenSustitúyalo por tu token de acceso de usuario. Para obtener información sobre cómo obtener tu token de acceso, consulta los Tokens de acceso de usuario en la documentación de Hugging Face.

   Para obtener más información sobre la huggingface_hub biblioteca, consulta la biblioteca cliente de Hub en la documentación de Hugging Face.

2. Para que el modelo descargado esté disponible para SageMaker AI, cárguelo en Amazon S3. En el siguiente ejemplo, se carga el modelo con el sagemaker_session objeto:

   custom_draft_model_uri = sagemaker_session.upload_data(
       path=hf_local_download_dir.as_posix(),
       bucket=sagemaker_session.default_bucket(),
       key_prefix="prefix",
   )

   En este ejemplo, prefix sustitúyalo por un calificador que le ayude a distinguir el borrador del modelo en S3, como. spec-dec-custom-draft-model

   El upload_data() método devuelve el URI de S3 de los artefactos del modelo.

3. Para ejecutar un trabajo de decodificación especulativo, utilice el optimize() método y establezca el speculative_decoding_config argumento. El siguiente ejemplo establece la ModelSource clave del URI de S3 del modelo de borrador personalizado:

   optimized_model = model_builder.optimize(
       instance_type="instance-type",
       accept_eula=True,
       speculative_decoding_config={
           "ModelSource": custom_draft_model_uri + "/",
       },
   )

   En este ejemplo, instance-type sustitúyala por una instancia de ML, comoml.p4d.24xlarge.

   El optimize() método devuelve un Model objeto, que puede utilizar para implementar el modelo en un punto final.

4. Cuando finalice el trabajo, despliegue el modelo. En el siguiente ejemplo se utiliza el deploy() método:

   predictor = optimized_model.deploy(accept_eula=True)

   El deploy() método devuelve un objeto predictor, que puede utilizar para enviar solicitudes de inferencia al punto final que aloja el modelo.

Para optimizar con la compilación

1. Para ejecutar un trabajo de compilación, utilice el optimize() método y establezca el compilation_config argumento. En el siguiente ejemplo, se utiliza la OverrideEnvironment clave para establecer las variables de entorno necesarias en el contenedor de optimización:

   optimized_model = model_builder.optimize(
       instance_type="instance-type",
       accept_eula=True,
       compilation_config={
           "OverrideEnvironment": {
               "OPTION_TENSOR_PARALLEL_DEGREE": "24",
               "OPTION_N_POSITIONS": "8192",
               "OPTION_DTYPE": "fp16",
               "OPTION_ROLLING_BATCH": "auto",
               "OPTION_MAX_ROLLING_BATCH_SIZE": "4",
               "OPTION_NEURON_OPTIMIZE_LEVEL": "2",
           }
       },
       output_path="s3://output-path",
   )

   En este ejemplo, instance-type configúrelo en un tipo de instancia de ML con hardware acelerado. Por ejemplo, para AWS una inferencia acelerada con Inferentia, puedes establecer el tipo en una instancia Inf2, como. ml.inf2.48xlarge s3://output-pathReemplácelo por la ruta a la ubicación S3 donde almacena el modelo optimizado que crea el trabajo.

2. Cuando se complete el trabajo, implemente el modelo. En el siguiente ejemplo se utiliza el deploy() método:

   predictor = optimized_model.deploy(accept_eula=True)

   El deploy() método devuelve un objeto predictor, que puede utilizar para enviar solicitudes de inferencia al punto final que aloja el modelo.

Prueba del modelo con una solicitud de inferencia

• Para enviar una solicitud de inferencia de prueba al modelo implementado, utilice el predict() método de un objeto predictor. En el siguiente ejemplo, se pasa la sample_input variable que también se pasó a la SchemaBuilder clase en los ejemplos para definir el modelo:

  predictor.predict(sample_input)

  La entrada de muestra tiene el mensaje,"What is the largest planet in the solar system?". El predict() método devuelve la respuesta que generó el modelo, como se muestra en el siguiente ejemplo:

  {'generated_text': ' Jupiter is the largest planet in the solar system. It is the fifth planet from the sun. It is a gas giant with . . .'}