Hub documentation

Uso de adaptadores en Hugging Face

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Uso de adaptadores en Hugging Face

Nota: Adapters ha reemplazado a la biblioteca adapter-transformers y es totalmente compatible en términos de pesos del modelo. Consulte aquí para más información.

Adapters es una biblioteca complementaria de 🤗 transformers para afinar de manera eficiente modelos de lenguaje preentrenados mediante adaptadores y otros métodos de eficiencia de parámetros. Adapters también proporciona varios métodos para la composición de módulos de adaptador durante el entrenamiento y la inferencia. Puedes aprender más sobre esto en el artículo de Adapters.

Explorar adaptadores en el Hub

Puedes encontrar modelos de Adapters filtrando a la izquierda de la página de modelos. Algunos modelos de adaptadores se encuentran en el repositorio de Adapter Hub. Los modelos de ambas fuentes se agregan en el sitio web de AdapterHub.

Instalación

Para comenzar, puedes consultar la guía de instalación de AdapterHub. También puedes usar la siguiente instalación de una línea a través de pip:

pip install adapters

Uso de modelos existentes

Para una guía completa sobre la carga de adaptadores preentrenados, recomendamos consultar la guía oficial.

Como resumen breve, una configuración completa consta de tres pasos:

  1. Cargue un modelo base de transformers con la clase AutoAdapterModel proporcionada por Adapters.
  2. Use el método load_adapter() para cargar y añadir un adaptador.
  3. Active el adaptador mediante active_adapters (para inferencia) o actívalo y establécelo como entrenable mediante train_adapter() (para entrenamiento). Asegúrate también de consultar composición de adaptadores.
from adapters import AutoAdapterModel

# 1.
model = AutoAdapterModel.from_pretrained("FacebookAI/roberta-base")
# 2.
adapter_name = model.load_adapter("AdapterHub/roberta-base-pf-imdb")
# 3.
model.active_adapters = adapter_name
# or model.train_adapter(adapter_name)

También puede usar list_adapters para encontrar todos los modelos de adaptador de forma programática:

from adapters import list_adapters

# source can be "ah" (AdapterHub), "hf" (hf.co) or None (for both, default)
adapter_infos = list_adapters(source="hf", model_name="FacebookAI/roberta-base")

Si quieres ver cómo cargar un modelo específico, puedes hacer clic en Use in Adapters y se te proporcionará un fragmento de código funcional para cargarlo.

Compartir tus modelos

Para una guía completa sobre el intercambio de modelos con Adapters, recomendamos consultar la guía oficial.

Puedes compartir tu adaptador usando el método push_adapter_to_hub desde un modelo que ya contiene un adaptador.

model.push_adapter_to_hub(
    "my-awesome-adapter",
    "awesome_adapter",
    adapterhub_tag="sentiment/imdb",
    datasets_tag="imdb"
)

Este comando crea un repositorio con una tarjeta de modelo generada automáticamente y todos los metadatos necesarios.

Recursos adicionales

Update on GitHub