Začněte nyníZačněte zdarma

Hledání nejpopulárnějšího modelu text-to-image

Teď si upřesníš hledání, aby sis našel/a a načetl/a nejpopulárnější model Stable Diffusion pro generování obrázků z textu od CompVis na Hugging Face Hubu.

Photo of a labrador chasing a ball

Hugging Face Hub API je już načtené (api), stejně jako modul StableDiffusionPipeline z knihovny diffusers. Pomocí modelů Stable Diffusion můžeš vygenerovat obrázek z libovolného promptu – třeba "a black labrador chasing a tennis ball".

Toto cvičení je součástí kurzu

Multi-Modal Models with Hugging Face

Zobrazit kurz

Pokyny k cvičení

  • Použij pouze modely určené pro úlohy text-to-image.
  • Přidej odpovídající tag, aby bylo zajištěno, že model půjde načíst třídou StableDiffusionPipeline z knihovny diffusers.
  • Seřaď výsledky podle počtu "likes".
  • Načti nejpopulárnější model ze seznamu models pomocí jeho ID.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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].____)
Upravit a spustit kód