Update README.md

README.md

@@ -27,46 +27,41 @@ base_model: tokyotech-llm/Swallow-MS-7b-v0.1
 # Usage
 
 ```python
-from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained(
-  "taoki/Swallow-MS-7b-v0.1-qlora-amenokaku-code"
+    "taoki/Swallow-MS-7b-v0.1-qlora-oaast1-jmulti-dolly-amenokaku"
 )
 model = AutoModelForCausalLM.from_pretrained(
-  "taoki/Swallow-MS-7b-v0.1-qlora-amenokaku-code"
+    "taoki/Swallow-MS-7b-v0.1-qlora-oaast1-jmulti-dolly-amenokaku"
 )
 
 if torch.cuda.is_available():
-  model = model.to("cuda")
+    model = model.to("cuda")
 
-prompt="""### Instruction:
-光の三原色は？
-### Response:
-"""
+prompt="[INST] 今日は4/1なのですが、周りの皆さんが突拍子もないことを言っていて困惑しています。一体何が起こっているのでしょうか？ [/INST]\n"
 
 input_ids = tokenizer(prompt, return_tensors="pt").to(model.device)
 outputs = model.generate(
-  **input_ids,
-  max_new_tokens=512,
-  do_sample=True,
-  top_p=0.95,
-  temperature=0.1,
-  repetition_penalty=1.0,
+    **input_ids,
+    max_new_tokens=512,
+    do_sample=True,
+    top_p=0.95,
+    temperature=0.1,
+    repetition_penalty=1.1,
 )
 print(tokenizer.decode(outputs[0]))
 ```
 
 # Output
 ````
-<s>### Instruction:
-光の三原色は？
-### Response:
- ```python
-print('赤')
-print('緑')
-print('青')
-```</s>
+<s> [INST] 今日は4/1なのですが、周りの皆さんが突拍子もないことを言っていて困惑しています。一体何が起こっているのでしょうか？ [/INST]
+4月1日はエイプリルフールで、人々は冗談やいたずらを言って楽しむ日とされています。この習慣は、1564年にフランスのシャルル9世が4月1日に結婚したことから始まったと言われている。
+
+しかし、あなたが困惑しているのなら、それはおそらく、あなたが冗談やいたずらを言っている人たちが、あなたがそのようなものだと思っていないことを言っているからだと思います。これは、彼らがあなたを驚かせたり、笑わせたりするために嘘をついている可能性があることを意味します。
+
+もし誰かがあなたを困惑させたり、不快にさせたりするようなことを言った場合は、気にせず無視してください。また、自分自身や他人を傷つけるような嘘をつくことは避けましょう。</s>
 ````
 
 This llama model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.