/ Модель

Характеристики

Контекст
1 024 токена
Лицензия
CreativeML OpenRAIL-M
Архитектура
gpt2 · dense
Опубликована
11 фев 2023

danbooru-tag-generator

Генерация текстаТекст → текст3 профессии

GPT-2, обученная на тегах Danbooru для генерации тегов и подсказок к аниме-изображениям.

287скачиваний за 30 дней20лайковвеса обновлены28 мар 2023

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Тесты пользователейпока нет

Никто ещё не опубликовал тест этой модели. Соберите бенчмарк из своих задач — результат появится здесь.

  1. Добавьте модель в бенчмарк

    Вместе с другими моделями категории «Текст» — сравнение идёт внутри одной категории.

  2. Опишите свои задачи

    Напишите запросы, при необходимости приложите файлы и эталонные ответы.

  3. Запустите и опубликуйте

    Модели ответят на каждую задачу, судья оценит ответы. Опубликованный результат появится на этой странице.

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

Danbooru Tag Generator — модель генерации текста на архитектуре GPT-2 от Fred Zhang. Она обучена в течение трёх эпох на 100 000 тегов изображений Danbooru с рейтингом up_score не ниже 3 и предназначена для генерации тегов и описаний в контексте аниме-арта. Модель связана с набором данных anime-prompts-180K; в репозитории доступны форматы PyTorch и ONNX, длина контекста — 1024 токена.

Модель может помогать составлять текстовые теги для аниме-изображений и формировать подсказки для генерации изображений. Карточка предупреждает, что исходный набор Danbooru не фильтрует материалы для взрослых, поэтому среди результатов могут встречаться NSFW-описания. Лицензия — CreativeML OpenRAIL-M.

Сценарии применения

  • генерация тегов для аниме-арта
  • составление текстовых описаний аниме-изображений
  • создание подсказок для генерации аниме-картинок
  • дополнение набора тегов Danbooru

Ограничения

Обучающие данные не фильтровались на NSFW-содержание, поэтому модель может генерировать теги с описаниями для взрослых. Контекст ограничен 1024 токенами.

Для каких профессий подходит3

Разделы каталога профессий, в задачах которых используется эта модель

Все профессии

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

Disclaimer

Danbooru stores millions of tagged anime images, but it doesn't have a way to filter out NSFW content. This model was trained on 100,000 of these tags with up_score ≥ 3 for 3 epochs, so it's possible that some tags might contain NSFW descriptions. So, just be mindful of that. Thank you for your cooperation.

The Safe Version

For details on data preprocessing, prompt engineering, and more, please see Fast Anime PromptGen. I used a very similar approach to train the Danbooru version.

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм

Можно запустить

Подбираем GPU…

Класс архитектуры
GPT2LMHeadModel
Ревизия весов
70b2842

0 моделей