Text-To-Pokémon

Основные возможности и характеристики

Модель дообучена на Stable Diffusion с использованием изображений Pokémon, снабжённых подписями BLIP, и умеет создавать изображения в стиле Pokémon по текстовым описаниям. Для создания персонажей не требуется сложная разработка промптов.

Использование и параметры ввода

• prompt (string): входной промпт, например Yoda.
• num_outputs (integer): количество выходных изображений; от 1 до 4, по умолчанию 1.
• num_inference_steps (integer): количество шагов удаления шума; от 1 до 50, по умолчанию 25.
• guidance_scale (number): масштаб classifier-free guidance; от 1 до 20, по умолчанию 7.5.
• seed (integer): случайное начальное значение; оставьте поле пустым для рандомизации.

Способы интеграции

1. Node.js / Python API: установите клиентскую библиотеку Replicate, задайте переменную окружения REPLICATE_API_TOKEN и запустите модель с хэшем версии ff6cc781634191dd3c49097a615d2fc01b0a8aae31c448e55039a04dcbf36bba, чтобы получить URL изображений или сохранить их на диске.
2. HTTP API: отправьте POST-запрос через curl на https://api.replicate.com/v1/predictions.
3. Cog: установите Cog и используйте cog predict r8.im/lambdal/text-to-pokemon@sha256:..., чтобы скачать и запустить модель локально.
4. Docker: запустите модель в локальной GPU-среде с помощью команды docker run и обращайтесь к API.

Целевая аудитория и основные преимущества

Модель предназначена для любителей AI-искусства, разработчиков игр и поклонников Pokémon. К основным преимуществам относятся специализированное дообучение для стиля Pokémon, тематически выразительные результаты генерации и поддержка локального развёртывания с открытым исходным кодом.

Цена и стоимость

Модель работает в Replicate и стоит примерно $0.047 за запуск (около 21 запуска за $1). Она работает на GPU Nvidia T4, а предсказания обычно выполняются в течение 4 минут. Модель имеет открытый исходный код и может бесплатно запускаться на локальном оборудовании.

Типичные варианты использования

Введите Yoda, чтобы создать образ мастера Йоды в стиле Pokémon; вводите Girl with a pearl earring, Donald Trump и другие описания, чтобы создавать забавных персонажей в стиле Pokémon.

История модели

Модель обучил Justin Pinkney в Lambda Labs, используя 2xA6000 GPU в Lambda GPU Cloud, примерно за 15 000 шагов. Обучение заняло около 6 часов и стоило около $10. Датасеты и веса опубликованы с открытым исходным кодом на Hugging Face.