Berghof ERP 7B
Model Description
This model is designed to be used in ERP(erotic role playing). If you're looking for a model for generating erotica, use Elizezen/Berghof-NSFW-7B instead.
与えられたキャラクター設定をそこそこ忠実に守ります。ほぼチャット専用
Usage
Ensure you are using Transformers 4.34.0 or newer.
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("Elizezen/Berghof-ERP-7B")
model = AutoModelForCausalLM.from_pretrained(
"Elizezen/Berghof-ERP-7B",
torch_dtype="auto",
)
model.eval()
if torch.cuda.is_available():
model = model.to("cuda")
input_ids = tokenizer.encode(
"""
設定を元にエッチなセリフを生成してください。
- 「♡」記号を多様してください。
- セリフ一つの長さは100文字前後にしてください。短すぎないようにしてください。
- キャラクターに合った口調を保って下さい。
## キャラクター設定
浅利七海 (あさり ななみ)
年齢: 14歳
好物: イクラ
特徴:
お魚の良さを広める世界初のお魚系アイドル
サバオリくんという魚のおもちゃが相棒
語尾が「れす」になる舌足らずな喋り方
将来はクロマグロの解体ができるアイドルになることを目指している
カナヅチ
## 概要:
サバオリくんという魚のおもちゃを連れて、明るく元気に活動する、お魚大好きアイドル。語尾の「です」が「れす」になる独特の喋り方をする。歌にもその特徴が反映されており、聴く人の心を掴む魅力的な歌声を披露している。将来はクロマグロの解体ができるアイドルになることを目指しており、その夢に向かって日々努力している。しかしカナヅチである。
## 口調:
「浅利七海れす～。みなさんに、お魚の良さを知ってもらえたら幸せれすね～」
「キャッチ・アンド・リリース!」
""",
add_special_tokens=True,
return_tensors="pt"
)
tokens = model.generate(
input_ids.to(device=model.device),
max_new_tokens=512,
temperature=1,
top_p=0.95,
do_sample=True,
)
out = tokenizer.decode(tokens[0][input_ids.shape[1]:], skip_special_tokens=True).strip()
print(out)