ComeçarComece de graça

Encontrando o modelo de texto para imagem mais popular

É hora de refinar sua busca para encontrar e carregar o modelo Stable Diffusion de texto para imagem mais popular em CompVis no Hugging Face Hub.

Foto de um labrador correndo atrás de uma bola

A API Hugging Face Hub foi carregada (api), assim como o módulo StableDiffusionPipeline da biblioteca diffusers. Com os modelos Stable Diffusion, você pode criar uma imagem a partir de qualquer prompt que quiser, tipo “um labrador preto correndo atrás de uma bola de tênis”.

Este exercício faz parte do curso

Modelos multimodais com Hugging Face

Ver curso

Instruções do exercício

  • Use só modelos que sejam para tarefas de text-to-image.
  • Use uma tag apropriada para garantir que o modelo possa ser carregado pela classe StableDiffusionPipeline da biblioteca diffusers.
  • Classifique os resultados de acordo com o número de "likes".
  • Carregue o modelo mais popular de models usando o ID dele.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

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].____)
Editar e executar o código