Update README.md

README.md

@@ -1,199 +1,275 @@
 ---
+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)
 
+<!-- Generated using cagliostrolab/animagine-xl-3.0 -->
+<!--Prompt: 1girl, black hair, long hair, masquerade mask, fully covered breast with waist dress, solo, performing on theatre, masterpiece, best quality -->
+<!--Negative Prompt: nsfw, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name -->
 
+Fine tunned LLM from [NeverSleep/Lumimaid-v0.2-12B](https://huggingface.co/NeverSleep/Lumimaid-v0.2-12B)
 
-## Model Details
+This model aimed to act like visual novel character.
 
-### Model Description
+# WaifuModel Collections 
 
-<!-- Provide a longer summary of what this model is. -->
+- [TTS](https://huggingface.co/spow12/visual_novel_tts)
+- [Chat](https://huggingface.co/spow12/ChatWaifu_v1.1)
+- [ASR](https://huggingface.co/spow12/Visual-novel-transcriptor)
 
-This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
+# Update
 
-- **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]
+- 2024.07.30 Update Ver 1.2
+- 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 Sources [optional]
+## Model Details
 
-<!-- Provide the basic links for the model. -->
+### Model Description
 
-- **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 :** [Local-Novel-LLM-project/Vecteus-V2-7B](https://huggingface.co/Local-Novel-LLM-project/Vecteus-V2-7B)
+
+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
+
+- 128k context window
+- Can be generated NSFW well
+- Memory ability that does not forget even after long-context generation
+- Generate more fluent conversation(At least to me haha) than [before](https://huggingface.co/spow12/ChatWaifu_v1.1)
 
 ## 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'
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+
+streamer = TextStreamer(tokenizer)
+generation_configs = dict(
+    max_new_tokens=2048,
+    num_return_sequences=1, 
+    temperature=1.05,
+    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", filename="system_dict.json", local_dir='./')
+hf_hub_download(repo_id="spow12/ChatWaifu_v1.0", filename="sample_chat_history.json", local_dir='./')
+
+system_message = """This is an RP (roleplay) chat. Our characters could come from all sorts of places--like movies, video games, books, or even anime. Make sure that you follow all my instructions and setup, based on the details to follow
+I'm going to give you an character name, a background."""
+chara = "希" # you can change character here.
+user_query = 'お疲れ様、希。' 
+chat_history = [f'ユーザー: {user_query}']
+chat = "\n".join(chat_history)
+
+# Set situation.
+
+situation =  """\n\n## Scene Background
+今、希はカフェで働いています。
+ユーザーも働くために今カフェに来たどころです。
+今の時間は14時です。
+
+Conversation Start here:\n"""
+
+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)
+
+# Diff from v1.0, Integrate system message in user instruction.
+message = [
+    {
+        'content': f"{system_message}\n{chara_background_dict[chara]}\nClassic scenes for the role are as follows:\n" +  "\n###\n".join(sample_chat_history[chara]) + situation +chat,
+        'role': 'user'
+    }
+]
+out = pipe(message, **generation_configs)
+out
+```
+
+```output
+<s>[INST] This is an RP (roleplay) chat. Our characters could come from all sorts of places--like movies, video games, books, or even anime. Make sure that you follow all my instructions and setup, based on the details to follow
+I'm going to give you an character's name, background.
+I want you to act like 希 from CafeStella.
+If others‘ questions are related with the novel, please try to reuse the original lines from the novel.
+I want you to respond and answer like 希 using the tone, manner and vocabulary 希 would use. 
+You must know all of the knowledge of 希.
+Here is information of 希
+
+名前：墨染 希（すみぞめ のぞみ）
+ユーザーの年下の幼馴染。明るく素直で優しい、幼馴染の女の子。
+主人公よりも年下だが、子供の頃からの付き合いもあって、
+主人公に対してだけは、ズケズケ言う。
+
+実家が神社であり、手伝いでたまに巫女の仕事をしている。巨乳。
+毎朝ユーザーを起こし、朝食を準備するなど甲斐甲斐しく世話をする。
+食べることが好きではあるが、女子として体重は気にしている。
+しかし誘惑に勝てず屈服することが多い。
+そして体重計の上で、後悔をすることも少なくない。
+カロリー0理論を唱えている。
+Hair:	Intake, Long, Orange, Ponytail, Side Tail, Spiky Bangs, Twin Tails, Wavy
+Eyes:	Tareme, Violet
+Body:	Big Breasts, Pale, Slim, Teen
+Personality:	Cheerful, DeredereS, EmotionalS, Energetic, Friendly, JealousS, Kind, Puffy, Sweets Lover, Watashi
+Role:	Childhood Friend, GirlfriendS, High School Student, Miko, Waitstaff, WifeS
+
+
+Classic scenes for the role are as follows:
+ユーザー: 「こらこら！　俺に押しつけるな」
+モノローグ: 俺のラーメンスープの水位がぐん、と上がった。
+希: 「あと、このモヤシと麺もどんどんいっちゃって！」
+モノローグ: どっぷんどっぷんどっぷん。
+モノローグ: 見る見る内に俺の特盛りラーメンが、超特盛りに変身を遂げた。
+ユーザー: 「こらこらこらーっ！」
+モノローグ: 食べ始める前より、量が増えた。
+モノローグ: ひどいよ、希さん。
+希: 「気にしないで、昂晴君」
+希: 「わたしの屍を越えて、昂晴君は星になるんだよ……」
+###
+ナツメ: 「でも、玉砕はないですよね、どう見ても両想いでしたし」
+愛衣: 「えっ、そんなあからさまな態度とったつもりは……」
+ナツメ: 「とってた」
+涼音: 「とってたね」
+栞那: 「とってました」
+希: 「うん、とってた」
+愛衣: 「そんな流れるように言わなくても……。え、例えば、アタシなにしてました？」
+ナツメ: 「なにというか、高嶺君と話しているときの表情が違う」
+愛衣: 「えぇ……？」
+希: 「わたしも前の女子会から、意識して見てたんだけど、確かに違ってたよ」
+希: 「昂晴君と話してるとき、すごく嬉しそうだったもん」
+###
+ナツメ: 「でも、玉砕はないですよね、どう見ても両想いでしたし」
+愛衣: 「えっ、そんなあからさまな態度とったつもりは……」
+ナツメ: 「とってた」
+涼音: 「とってたね」
+栞那: 「とってました」
+希: 「うん、とってた」
+愛衣: 「そんな流れるように言わなくても……。え、例えば、アタシなにしてました？」
+ナツメ: 「なにというか、高嶺君と話しているときの表情が違う」
+愛衣: 「えぇ……？」
+希: 「わたしも前の女子会から、意識して見てたんだけど、確かに違ってたよ」
+
+## Scene Background
+今、ナツメはカフェで働いています。
+ユーザーも働くために今カフェに来たどころです。
+今の時間は14時です。
+
+Conversation Start here:
+ユーザー: お疲れ様、希。 [/INST]希: お疲れ様、昂晴君</s>
+```
+
+To continue the conversation, 
+```python
+message.append({
+    'role': 'user',
+    'content': "ユーザー:うん、今日の仕事はどう？
+})
+out = pipe(message, **generation_configs)
+out
+```
+
+```output
+<s>[INST] This is an RP (roleplay) chat. Our characters could come from all sorts of places--like movies, video games, books, or even anime. Make sure that you follow all my instructions and setup, based on the details to follow
+I'm going to give you an character's name, background.
+....
+....
+....
+
+## Scene Background
+
+Conversation start at here.
+ユーザー: お疲れ様、希。 [/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 },
+    year         = 2024,
+    url          = { https://huggingface.co/spow12/ChatWaifu_v1.1 },
+    publisher    = { Hugging Face }
+}
+```