Hub documentation

Utilisation des adaptateurs sur Hugging Face

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Utilisation des adaptateurs sur Hugging Face

Note : Adapters a remplacé la bibliothèque adapter-transformers et est entièrement compatible en termes de poids de modèle. Voir ici pour plus de détails.

Adapters est une bibliothèque complémentaire de 🤗 transformers permettant d’affiner efficacement des modèles de langage pré-entraînés à l’aide d’adapters et d’autres méthodes à efficacité paramétrique. Adapters fournit également diverses méthodes pour la composition de modules d’adapter pendant l’entraînement et l’inférence. Vous pouvez en savoir plus à ce sujet dans l’article Adapters.

Explorer les adaptateurs sur le Hub

Vous pouvez trouver des modèles Adapters en filtrant à gauche de la page des modèles. Certains modèles d'adaptateurs sont disponibles dans le dépôt Adapter Hub. Les modèles des deux sources sont agrégés sur le site web AdapterHub.

Installation

Pour commencer, vous pouvez consulter le guide d'installation d'AdapterHub. Vous pouvez également utiliser l'installation en une ligne suivante via pip :

pip install adapters

Utilisation de modèles existants

Pour un guide complet sur le chargement d'adapteurs pré-entraînés, nous vous recommandons de consulter le guide officiel.

En résumé, une configuration complète se déroule en trois étapes :

  1. Chargez un modèle de base transformers avec la classe AutoAdapterModel fournie par Adapters.
  2. Utilisez la méthode load_adapter() pour charger et ajouter un adaptateur.
  3. Activez l’adaptateur via active_adapters (pour l’inférence) ou activez-le et définissez-le comme entraînable via train_adapter() (pour l’entraînement). Assurez-vous également de consulter la composition des adaptateurs.
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)

Vous pouvez également utiliser list_adapters pour trouver tous les modèles d'adaptateurs de manière programmatique :

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 vous souhaitez voir comment charger un modèle spécifique, vous pouvez cliquer sur Use in Adapters et un extrait de code fonctionnel vous sera fourni pour le charger !

Partager vos modèles

Pour un guide complet sur le partage de modèles avec Adapters, nous vous recommandons de consulter le guide officiel.

Vous pouvez partager votre adaptateur en utilisant la méthode push_adapter_to_hub depuis un modèle qui contient déjà un adaptateur.

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

Cette commande crée un dépôt avec une carte de modèle générée automatiquement et toutes les métadonnées nécessaires.

Ressources supplémentaires

Update on GitHub