/ Модель

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

Контекст
1 024 токена
Архитектура
gpt2 · dense

gpt2-genre-story-generator

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

Дообученная на жанровых рассказах GPT-2 генерирует истории по жанру и текстовой затравке.

238скачиваний за 30 дней45лайковвеса обновлены23 мая 2021

Бенчмарки

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

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

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

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

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

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

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

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

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

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

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

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

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

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

GPT-2, дообученная автором карточки на рассказах разных жанров для генерации историй по жанру и начальному фрагменту. Модель принимает текстовую подсказку; среди поддерживаемых жанров указаны супергеройский, боевик, драма, ужасы, триллер и научная фантастика. В README приведён пример использования через библиотеку Transformers.

Модель основана на предварительно обученном чекпойнте gpt2 и дообучена для генерации текста. Архитектура — плотная GPT-2 с классом GPT2LMHeadModel; доступны веса PyTorch и Flax, контекст составляет 1024 токена. Языки рассказов и условия лицензии в предоставленных данных не описаны.

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

  • создание начала рассказа по заданному жанру
  • продолжение истории после начальной фразы
  • генерация сюжетных набросков в жанре супергероики
  • создание черновиков фантастических или драматических рассказов

Ограничения

Контекст ограничен 1024 токенами; качество и языковая поддержка не охарактеризованы в предоставленных данных.

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

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

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

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

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

Открыть на Hugging Face

GPT2 Genre Based Story Generator

Model description

GPT2 fine-tuned on genre-based story generation.

Intended uses

Used to generate stories based on user inputted genre and starting prompts.

How to use

Supported Genres

superhero, action, drama, horror, thriller, sci_fi

Input text format

<BOS> <genre> Some optional text...

Example: <BOS> <sci_fi> After discovering time travel,

# Example of usage
from transformers import pipeline

story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
print(story_gen("<BOS> <superhero> Batman"))

Training data

Initialized with pre-trained weights of "gpt2" checkpoint. Fine-tuned the model on stories of various genres.

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей