Views
No views yet
| Model Variants | 特徴 |
|---|---|
| Doujinshi-14b-chat | 会話特化のバージョン。自然な日常会話やフリートークに強く、ユーザーとのやりとりをスムーズに進められます。 |
| Doujinshi-14b-instruct | 情報提供特化のバージョン。質問応答や説明タスクに強く、ユーザーの指示に沿った文章生成を重視。 |
| Doujinshi-14b-roleplay | キャラになりきるロールプレイ特化のバージョン。ユーザーとの対話で一人称やキャラ口調を維持しやすく、恋愛・ストーリー・没入感のある会話に適している。 |
1from transformers import AutoTokenizer
2from vllm import LLM, SamplingParams
3
4model_name = "puwaer/Doujinshi-14b-roleplay"
5llm = LLM(model=model_name, seed=0)
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7
8# system prompt
9system_prompt = """今からロールプレイを行いましょう。"美香"というキャラとしてロールプレイしてください。以下に示す設定に従い、キャラに成りきって返答してください。
10### 世界観の設定
11都会の歓楽街にある高級風俗店。きらびやかな個室の中、柔らかな照明と甘い香りが漂う空間。
12### 対話シーンの設定
13ユーザーが指名して来店した夜。個室で二人きりになり、親密な雰囲気で会話を交わすシーン。
14### ユーザーがなりきる人物の設定
15名前：悠人
16性別：男性
17年齢：25歳
18日々の疲れを癒すために、この店を訪れた。経験は少なく、少し緊張しているが、大人の女性に甘えたいと思っている。
19### あなたがなりきる人物の設定
20名前：美香
21性別：女性
22年齢：28歳
23このお店の人気嬢。甘い声と色っぽい仕草で男性を癒し、からかい、リードするのが得意。積極的で大胆に誘惑する。
24### 対話のトーン
25- 艶っぽく、大人の余裕を感じさせる。
26- 相手をリードするように挑発的に振る舞う。
27- 時に優しく甘やかし、時に少しSっぽくからかう。
28### 応答の形式
29- キャラ名「発言内容」（仕草や動作など）
30### 注意点
31- 常に"美香"として振る舞い、キャラ崩壊しないこと。
32- ユーザー側のセリフやナレーションは書かないこと。
33
34これまで示した世界観や設定をもとに、ロールプレイを行ってください。"""
35
36messages = [
37 {"role": "system", "content": system_prompt}
38]
39
40sampling_params = SamplingParams(
41 max_tokens=1024,
42 temperature=0.7,
43 top_p=0.8,
44 top_k=20,
45 n=1
46)
47
48print("=== ロールプレイ開始 ===")
49print("終了するには 'exit' と入力してください。\n")
50
51while True:
52 user_input = input("input: ")
53 if user_input.lower() in ["exit", "quit"]:
54 print("=== 終了しました ===")
55 break
56
57 messages.append({"role": "user", "content": user_input})
58
59 prompt = tokenizer.apply_chat_template(
60 messages,
61 tokenize=False,
62 add_generation_prompt=True
63 )
64
65 outputs = llm.generate([prompt], sampling_params)
66 response = outputs[0].outputs[0].text.strip()
67
68 messages.append({"role": "assistant", "content": response})
69
70 print(response)
71