ПочатиПочніть безкоштовно

Пошук найпопулярнішої моделі text-to-image

Час звузити пошук, щоб знайти та завантажити найпопулярнішу модель Stable Diffusion для перетворення тексту на зображення від CompVis у Hugging Face Hub.

Photo of a labrador chasing a ball

API Hugging Face Hub уже завантажено (api), як і модуль StableDiffusionPipeline з бібліотеки diffusers. За допомогою моделей Stable Diffusion ви можете згенерувати зображення за будь-яким промптом, наприклад: "a black labrador chasing a tennis ball".

Ця вправа є частиною курсу

Багатомодальні моделі з Hugging Face

Переглянути курс

Інструкції до вправи

  • Використовуйте лише моделі для завдання text-to-image.
  • Додайте відповідний тег, щоб модель можна було завантажити класом StableDiffusionPipeline з бібліотеки diffusers.
  • Відсортуйте результати за кількістю "likes".
  • Завантажте найпопулярнішу модель з models, використавши її ID.

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

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].____)
Редагувати та запускати код