spow12 commited on
Commit
3947c92
1 Parent(s): e6a76e5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -0
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