l_soft_erotic / README.md
zlsl's picture
Update README.md
893ce89
metadata
license: cc-by-nc-sa-4.0
language:
  - ru
library_name: transformers
tags:
  - erotic
  - porn
  - sexstory
  - pornstory
widget:
  - text: Нежная кожа
  - text: Укусил за

Модель для генерации легкой эротики Неплохо держит конекст, параметры для старта:

do_sample: true
top_p: 0.96
top_k: 14
temperature: 0.88 # Выше 1.2 - будет сильно уходить от контекста
repetition_penalty: 1.16
encoder_repetition_penalty: 1.09-1.15
typical_p: 1.0

Для пользователей text-generation-webui

В инструменте поломана работа с GPT-2, GPTJ, GPT-NEO и аналогичными модлями, неверно загружается токенизер. Ошибка такая:

eos_token_id = eos_token_id[0] IndexError: list index out of range

Исправляется легко, в файл modules/models.py в функцию load_tokenizer() надо добавить строчку
tokenizer.eos_token_id = 2
перед
return tokenizer