Update README.md
Browse files
README.md
CHANGED
@@ -44,6 +44,7 @@ character | visual_novel |
|
|
44 |
|
45 |
```python
|
46 |
from transformers import TextStreamer, pipeline, AutoTokenizer, AutoModelForCausalLM
|
|
|
47 |
|
48 |
model_id = 'spow12/Roleplayer-VisualNovel'
|
49 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
@@ -77,6 +78,10 @@ user_query = '「おはいよ、ムラサメ。」'
|
|
77 |
chara = "ムラサメ"
|
78 |
chat_history = [f'ユーザー: {user_query}']
|
79 |
chat = "\n".join(chat_history)
|
|
|
|
|
|
|
|
|
80 |
|
81 |
message = [
|
82 |
{
|
@@ -146,6 +151,8 @@ This model is currently available for non-commercial & Research purpose only. Al
|
|
146 |
|
147 |
By sharing this model, I hope to contribute to the research efforts of our community (the open-source community and anime persons).
|
148 |
|
|
|
|
|
149 |
|
150 |
## Citation
|
151 |
|
|
|
44 |
|
45 |
```python
|
46 |
from transformers import TextStreamer, pipeline, AutoTokenizer, AutoModelForCausalLM
|
47 |
+
import json
|
48 |
|
49 |
model_id = 'spow12/Roleplayer-VisualNovel'
|
50 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
|
|
78 |
chara = "ムラサメ"
|
79 |
chat_history = [f'ユーザー: {user_query}']
|
80 |
chat = "\n".join(chat_history)
|
81 |
+
# Note you have to change the path of system message dict.
|
82 |
+
# Check the repository file.
|
83 |
+
with open('system_dict.json', 'r') as f:
|
84 |
+
chara_background_dict = json.load(f)
|
85 |
|
86 |
message = [
|
87 |
{
|
|
|
151 |
|
152 |
By sharing this model, I hope to contribute to the research efforts of our community (the open-source community and anime persons).
|
153 |
|
154 |
+
This repository can use Visual novel-based RAG, but i will not distribute it yet because i'm not sure if it is permissible to release the data publicly.
|
155 |
+
|
156 |
|
157 |
## Citation
|
158 |
|