ÎncepețiÎncepe gratuit

Găsirea celui mai popular model text-to-image

E momentul să-ți rafinezi căutarea pentru a găsi și încărca cel mai popular model Stable Diffusion de tip text-to-image de la CompVis de pe Hugging Face Hub.

Photo of a labrador chasing a ball

API-ul Hugging Face Hub a fost încărcat (api), la fel și modulul StableDiffusionPipeline din biblioteca diffusers. Cu modelele Stable Diffusion poți genera o imagine din orice prompt dorești, de exemplu: „un labrador negru care aleargă după o minge de tenis".

Acest exercițiu face parte din cursul

Modele Multi-Modale cu Hugging Face

Vezi cursul

Instrucțiuni pentru exercițiu

  • Folosește doar modele destinate sarcinilor de tip text-to-image.
  • Folosește un tag potrivit pentru a te asigura că modelul poate fi încărcat de clasa StableDiffusionPipeline din biblioteca diffusers.
  • Sortează rezultatele după numărul de "likes".
  • Încarcă cel mai popular model din models folosind ID-ul său.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

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].____)
Editează și rulează codul