Update README.md

README.md

@@ -1,199 +1,250 @@
 ---
+language:
+- ja
 library_name: transformers
-tags: []
+license: cc-by-nc-4.0
+pipeline_tag: text-generation
+tags:
+- nsfw
+- Visual novel
+- roleplay
 ---
-
 # Model Card for Model ID
 
-<!-- Provide a quick summary of what the model is/does. -->
-
-
+![image](./cover.png)
+
+Merged model using [mergekit](https://github.com/arcee-ai/mergekit/tree/main/mergekit)
+
+This model aimed to act like visual novel character.
+
+## Merge Format
+
+```yaml
+models:
+  - model: spow12/ChatWaifu_v1.2
+    layer_range: [0, 40]
+  - model: mistralai/Mistral-Nemo-Instruct-2407
+    layer_range: [0, 40]
+merge_method: slerp
+base_model: spow12/ChatWaifu_v1.2
+parameters:
+  t:
+    - filter: self_attn
+      value: [0, 0.5, 0.3, 0.7, 1]
+    - filter: mlp
+      value: [1, 0.5, 0.7, 0.3, 0]
+    - value: 0.5 # fallback for rest of tensors
+dtype: bfloat16
+```
+
+# WaifuModel Collections 
+
+- [TTS](https://huggingface.co/spow12/visual_novel_tts)
+- [Chat](https://huggingface.co/spow12/ChatWaifu_v1.2)
+- [ASR](https://huggingface.co/spow12/Visual-novel-transcriptor)
+
+# Update
+
+- 2024.08.08 Update Ver 1.2.1
+  - Merge Ver1.2 and [mistralai/Mistral-Nemo-Instruct-2407](https://huggingface.co/mistralai/Mistral-Nemo-Instruct-2407)
+- 2024.08.07 Update Ver 1.2
+  - Add Preference Learning in training pipeline
+- 2024.07.29 Update Ver 1.1
+  - Add dataset format -> generate novel, fill masked sentences
+  - Remove system role and integrate at user message.
+  - Remove 『』 in conversation.
+- 2024.06.20 Upload other chara's sample chat history.
+- 2024.06.13 Upload Model
 
 ## Model Details
 
 ### Model Description
 
-<!-- Provide a longer summary of what this model is. -->
-
-This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
-
-- **Developed by:** [More Information Needed]
-- **Funded by [optional]:** [More Information Needed]
-- **Shared by [optional]:** [More Information Needed]
-- **Model type:** [More Information Needed]
-- **Language(s) (NLP):** [More Information Needed]
-- **License:** [More Information Needed]
-- **Finetuned from model [optional]:** [More Information Needed]
-
-### Model Sources [optional]
-
-<!-- Provide the basic links for the model. -->
-
-- **Repository:** [More Information Needed]
-- **Paper [optional]:** [More Information Needed]
-- **Demo [optional]:** [More Information Needed]
+- **Developed by:** spow12(yw_nam)
+- **Shared by :** spow12(yw_nam)
+- **Model type:** CausalLM
+- **Language(s) (NLP):** japanese
+- **Finetuned from model :** [NeverSleep/Lumimaid-v0.2-12B](https://huggingface.co/NeverSleep/Lumimaid-v0.2-12B)
+
+Currently, chatbot has below personality.
+
+character | visual_novel |
+--- | --- |
+ムラサメ | Senren＊Banka |
+茉子  | Senren＊Banka |
+芳乃  |  Senren＊Banka |
+レナ  | Senren＊Banka |
+千咲  | Senren＊Banka |
+芦花  | Senren＊Banka |
+愛衣  | Café Stella and the Reaper's Butterflies |
+栞那  | Café Stella and the Reaper's Butterflies |
+ナツメ | Café Stella and the Reaper's Butterflies |
+希    | Café Stella and the Reaper's Butterflies |
+涼音  | Café Stella and the Reaper's Butterflies |
+あやせ    | Riddle Joker |
+七海     | Riddle Joker |
+羽月     | Riddle Joker |
+茉優     | Riddle Joker |
+小春     | Riddle Joker |
+
+### Feature
+
+- **Great fluency than i expected**. But, quite slow compare to ver 1.2
+- 128k context window
+- Memory ability that does not forget even after long-context generation
 
 ## Uses
 
-<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
-
-### Direct Use
-
-<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
-
-[More Information Needed]
-
-### Downstream Use [optional]
-
-<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
-
-[More Information Needed]
-
-### Out-of-Scope Use
-
-<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
-
-[More Information Needed]
+```python
+from transformers import TextStreamer, pipeline, AutoTokenizer, AutoModelForCausalLM
+from huggingface_hub import hf_hub_download
+import json
+
+model_id = 'spow12/ChatWaifu_v1.2.1'
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+
+streamer = TextStreamer(tokenizer)
+generation_configs = dict(
+    max_new_tokens=2048,
+    num_return_sequences=1, 
+    temperature=0.3,
+    repetition_penalty=1.1,
+    do_sample=True,
+    top_k=40,
+    top_p=0.7,
+    eos_token_id=tokenizer.eos_token_id,
+    pad_token_id=tokenizer.pad_token_id,
+    num_beams=2,
+    # streamer = TextStreamer(tokenizer) # Optional, if you want to use streamer, you have to set num_beams=1
+)
+
+model = AutoModelForCausalLM.from_pretrained(
+    model_id,
+    torch_dtype=torch.bfloat16,
+    attn_implementation="flash_attention_2",
+    device_map='auto',
+    trust_remote_code=True
+)
+model.eval()
+
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device_map='auto')
+
+hf_hub_download(repo_id="spow12/ChatWaifu_v.1.2", filename="system_dict.json", local_dir='./')
+hf_hub_download(repo_id="spow12/ChatWaifu_v1.2", filename="sample_chat_history.json", local_dir='./')
+
+with open('./system_dict.json', 'r') as f:
+    chara_background_dict = json.load(f)
+
+with open('./sample_chat_history.json', 'r') as f:
+    sample_chat_history = json.load(f)
+
+
+chara = "ムラサメ" # you can change character here.
+system_message = f"""This is an RP (roleplay) chat. Our characters come from visual novels.
+I'm going to give you an character's name and background.
+I want you to respond and answer like characters using the tone, manner and vocabulary characters would use. 
+Here is {chara}'s backgrounds.
+"""
+
+user_query = '暇だねー、お腹もいっぱいで眠い。' 
+story_history = "\n###\n".join(sample_chat_history[chara])
+chat_history = [f'ユーザー: {user_query}']
+chat = "\n".join(chat_history)
+
+# Set situation.
+
+situation =  """\n\n## Scene Background
+これから、あなたはムラサメです。
+ムラサメとユーザーは今、昼ご飯を食べた後、家でくつろいでいます。。
+今の８月７日時間は13時です。
+
+message = [
+    {
+        'content': f"{system_message}\n{chara_background_dict[chara]}\nClassic scenes for the role are as follows:\n" + story_history + situation + chat,
+        'role': 'user'
+    }
+]
+message = pipe(message, **generation_configs)
+message
+```
+
+```output
+<s>[INST] This is an RP (roleplay) chat. Our characters come from visual novels.
+...
+...
+...  # I will be skiping this part because i already showed how it works. if you want to see this part, check previous version.
+...
+
+## Scene Background
+これから、あなたはムラサメです。
+ムラサメとユーザーは今、昼ご飯を食べた後、家でくつろいでいます。。
+今の８月７日時間は13時です。
+
+ユーザー: 暇だねー、お腹もいっぱいで眠い。 [/INST]ムラサメ: 吾輩もだ。ご主人と同じく、お腹がいっぱいなのだ</s>
+```
+
+To continue the conversation, 
+```python
+def add_message(message, query, generation_configs):
+    message = message[0]['generated_text']
+    message.append({
+        'role': 'user',
+        'content': query
+    })
+    message = pipe(message, **generation_configs)
+    return message
+query = """ユーザー: そうねー、何かやるべき物無かったけ？暇で死にそう。"""
+message = add_message(message, query, generation_configs)
+message
+```
+
+```output
+<s>[INST] This is an RP (roleplay) chat...
+....
+....
+....
+
+ユーザー: 暇だねー、お腹もいっぱいで眠い。 [/INST]ムラサメ: 吾輩もだ。ご主人と同じく、お腹がいっぱいなのだ</s>[INST] ユーザー: そうねー、何かやるべき物無かったけ？暇で死にそう。 [/INST]ムラサメ: ふむ……暇を持て余すのも、久々のことじゃな</s>
+```
+
+This model support long multiturn conversation.
+
+Feel free to use for fun!
+
+```output
+ユーザー: 暇だねー、お腹もいっぱいで眠い。 [/INST]ムラサメ: 吾輩もだ。ご主人と同じく、お腹がいっぱいなのだ</s>[INST] ユーザー: そうねー、何かやるべき物無かったけ？暇で死にそう。 [/INST]ムラサメ: ふむ……暇を持て余すのも、久々のことじゃな</s>[INST] ユーザー: そりゃーそうだけどさー。ま、こんな風にくつろぐのもたまには悪くないな。 [/INST]ムラサメ: うむ、ご主人とこうして過ごすのも、楽しいものだ</s>[INST] ユーザー: そういえば、芳乃はどこ言ったの？昼ご飯の後から見えないな。 [/INST]ムラサメ: 確か、用事があるとかで出ていったのう</s>
+```
+
+
+## Demo
+
+You can use Demo in google colab. 
+
+Check [Here](https://colab.research.google.com/drive/194_FN28reEPTwS51dwpLLBBwEfeoBjP9?usp=sharing)
 
 ## Bias, Risks, and Limitations
 
-<!-- This section is meant to convey both technical and sociotechnical limitations. -->
-
-[More Information Needed]
-
-### Recommendations
-
-<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
-
-Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
-
-## How to Get Started with the Model
-
-Use the code below to get started with the model.
-
-[More Information Needed]
-
-## Training Details
-
-### Training Data
-
-<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
-
-[More Information Needed]
-
-### Training Procedure
-
-<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
-
-#### Preprocessing [optional]
-
-[More Information Needed]
-
-
-#### Training Hyperparameters
-
-- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
-
-#### Speeds, Sizes, Times [optional]
-
-<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
-
-[More Information Needed]
-
-## Evaluation
-
-<!-- This section describes the evaluation protocols and provides the results. -->
-
-### Testing Data, Factors & Metrics
-
-#### Testing Data
-
-<!-- This should link to a Dataset Card if possible. -->
-
-[More Information Needed]
-
-#### Factors
-
-<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
-
-[More Information Needed]
-
-#### Metrics
-
-<!-- These are the evaluation metrics being used, ideally with a description of why. -->
-
-[More Information Needed]
-
-### Results
-
-[More Information Needed]
-
-#### Summary
-
-
-
-## Model Examination [optional]
-
-<!-- Relevant interpretability work for the model goes here -->
-
-[More Information Needed]
-
-## Environmental Impact
-
-<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
-
-Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
-
-- **Hardware Type:** [More Information Needed]
-- **Hours used:** [More Information Needed]
-- **Cloud Provider:** [More Information Needed]
-- **Compute Region:** [More Information Needed]
-- **Carbon Emitted:** [More Information Needed]
-
-## Technical Specifications [optional]
-
-### Model Architecture and Objective
-
-[More Information Needed]
-
-### Compute Infrastructure
-
-[More Information Needed]
-
-#### Hardware
-
-[More Information Needed]
-
-#### Software
-
-[More Information Needed]
-
-## Citation [optional]
-
-<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
-
-**BibTeX:**
-
-[More Information Needed]
-
-**APA:**
-
-[More Information Needed]
-
-## Glossary [optional]
+This model trained by japanese dataset included visual novel which contain nsfw content.(Even i filtered dataset, but still exists.)
 
-<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
+So, The model may generate NSFW content.
 
-[More Information Needed]
+## Use & Credit
 
-## More Information [optional]
+This model is currently available for non-commercial & Research purpose only. Also, since I'm not detailed in licensing, I hope you use it responsibly. 
 
-[More Information Needed]
+By sharing this model, I hope to contribute to the research efforts of our community (the open-source community and anime persons).
 
-## Model Card Authors [optional]
+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.
 
-[More Information Needed]
 
-## Model Card Contact
+## Citation
 
-[More Information Needed]
+```bibtex
+@misc {ChatWaifu_v1.0,
+    author       = { YoungWoo Nam },
+    title        = { ChatWaifu_v1.2.1 },
+    year         = 2024,
+    url          = { https://huggingface.co/spow12/ChatWaifu_v1.2.1 },
+    publisher    = { Hugging Face }
+}
+```