CommencezCommencez gratuitement

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

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

Photo d'un labrador qui court après une balle

L'API du Hub Hugging Face a été chargée (api), tout comme 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 invite, par exemple « a black labrador chasing a tennis ball ».

Cette activité fait partie du cours

Modèles multimodaux avec Hugging Face

Voir le cours

Instructions de l’exercice

  • N'utilisez que des modèles pour les tâches text-to-image.
  • Utilisez une étiquette appropriée pour vous assurer 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 à partir de models en utilisant son identifiant.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

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