Rechercher sur le Hub

Dans ce tutoriel, vous apprendrez à rechercher des modèles, des jeux de données et des espaces sur le Hub à l'aide de huggingface_hub.

Comment lister les dépôts ?

La bibliothèque huggingface_hub inclut un client HTTP HfApi pour interagir avec le Hub. Entre autres, il peut lister les modèles, les jeux de données et les espaces stockés sur le Hub :

>>> from huggingface_hub import HfApi
>>> api = HfApi()
>>> models = api.list_models()

La sortie de list_models() est un itérateur sur les modèles stockés sur le Hub.

De même, vous pouvez utiliser list_datasets() pour lister les jeux de données et list_spaces() pour lister les Spaces.

Comment filtrer les dépôts ?

Lister les dépôts est pratique, mais vous voudrez peut-être maintenant filtrer votre recherche. Les assistants de liste possèdent plusieurs attributs tels que :

• filter
• author
• search
• num_parameters
• …

Voici un exemple pour obtenir tous les modèles du Hub qui effectuent la classification d’images, qui ont été entraînés sur le jeu de données imagenet et qui fonctionnent avec PyTorch.

models = hf_api.list_models(filter=["image-classification", "pytorch", "imagenet"])

Vous pouvez également filtrer les modèles par nombre de paramètres en utilisant la même syntaxe d’intervalle que l’interface du Hub :

models = hf_api.list_models(num_parameters="min:6B,max:128B")

Lors du filtrage, vous pouvez également trier les modèles et ne conserver que les meilleurs résultats. Par exemple, l'exemple suivant récupère les 5 jeux de données les plus téléchargés sur le Hub :

>>> list(list_datasets(sort="downloads", limit=5))
[DatasetInfo(
	id='argilla/databricks-dolly-15k-curated-en',
	author='argilla',
	sha='4dcd1dedbe148307a833c931b21ca456a1fc4281',
	last_modified=datetime.datetime(2023, 10, 2, 12, 32, 53, tzinfo=datetime.timezone.utc),
	private=False,
	downloads=8889377,
	(...)

Pour explorer les filtres disponibles sur le Hub, visitez les pages models et datasets dans votre navigateur, recherchez certains paramètres et examinez les valeurs dans l’URL.

Utilisation de l'interface en ligne de commande

Vous pouvez également lister et rechercher des modèles, des jeux de données et des Spaces à l’aide de l’interface en ligne de commande hf :

# List models
>>> hf models ls --search "llama" --sort downloads --limit 5

# List datasets
>>> hf datasets ls --author Qwen

# List Spaces
>>> hf spaces ls --search "3d"

# Get info about a specific model
>>> hf models info Lightricks/LTX-2

Pour plus de détails, consultez le guide CLI.