zlsl's picture
Update README.md
117da12
metadata
license: cc-by-nc-sa-4.0
language:
  - ru
library_name: transformers
pipeline_tag: text-generation
tags:
  - erotic
  - russian
  - chat
  - roleplay
widget:
  - text: 'Я: (улыбаюсь)'
  - text: 'Аня: Ох!'

float16 версия zlsl/l_soft_erotic_tm

Для пользователей 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