CommencerCommencer gratuitement

Trouver le modèle texte‑vers‑image le plus populaire

Il est temps d’affiner votre recherche pour trouver et charger sur le Hub Hugging Face le modèle Stable Diffusion texte‑vers‑image le plus populaire de CompVis.

Photo of a labrador chasing a ball

L’API du Hub Hugging Face a été chargée (api), ainsi que le module StableDiffusionPipeline de la bibliothèque diffusers. Avec les modèles Stable Diffusion, vous pouvez générer une image à partir de n’importe quel prompt, par exemple : "a black labrador chasing a tennis ball".

Cet exercice fait partie du cours

Modèles multimodaux avec Hugging Face

Afficher le cours

Instructions

  • N’utilisez que des modèles pour la tâche text-to-image.
  • Utilisez une balise appropriée pour garantir que le modèle peut être chargé par la classe StableDiffusionPipeline de la bibliothèque diffusers.
  • Triez les résultats selon le nombre de "likes".
  • Chargez le modèle le plus populaire de models en utilisant son identifiant.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

models = api.list_models(
    # Filter for text-to-image tasks
    task="____",
    author="CompVis",
    # Filter for models that can be loaded by the diffusers library
    tags="diffusers:____",
    # Sort according to the most popular
    sort="____"
)

models = list(models)

# Load the most popular model from models
pipe = StableDiffusionPipeline.from_pretrained(models[0].____)
Modifier et exécuter le code