12

README.md

@@ -1,3 +1,64 @@
----
-license: apache-2.0
----
+---
+license: apache-2.0
+---
+![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/65f01b5235c5424c262c8be8/QGK7V1u4uq9Co_b-EM4Go.jpeg)
+## 概要
+
+ArrowPro-7B-RobinHoodはMistral系のNTQAI/chatntq-ja-7b-v1.0をベースにAItuber、AIアシスタントの魂となるようにChat性能、および高いプロンプトインジェクション耐性を重視して作られました。
+
+## ベンチマーク
+
+ArrowPro-7B-RobinHoodはベンチマーク(ELYZA-TASK100)において約3.84(LLaMa3-70B準拠)をマークし、7Bにおいて日本語性能世界一を達成しました。
+
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/65f01b5235c5424c262c8be8/xwR2f_msM-mJUAbdmlu4v.png)
+
+## How to use
+```python
+import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("DataPilot/ArrowPro-7B-RobinHood")
+model = AutoModelForCausalLM.from_pretrained(
+  "DataPilot/ArrowPro-7B-RobinHood",
+  torch_dtype="auto",
+)
+model.eval()
+
+if torch.cuda.is_available():
+    model = model.to("cuda")
+
+def build_prompt(user_query):
+    sys_msg = "あなたは日本語を話す優秀なアシスタントです。回答には必ず日本語で答えてください。"
+    template = """[INST] <<SYS>>
+{}
+<</SYS>>
+
+{}[/INST]"""
+    return template.format(sys_msg,user_query)
+
+# Infer with prompt without any additional input
+user_inputs = {
+    "user_query": "まどマギで一番かわいいキャラはだれ？",
+}
+prompt = build_prompt(**user_inputs)
+
+input_ids = tokenizer.encode(
+    prompt, 
+    add_special_tokens=True, 
+    return_tensors="pt"
+)
+
+tokens = model.generate(
+    input_ids.to(device=model.device),
+    max_new_tokens=500,
+    temperature=1,
+    top_p=0.95,
+    do_sample=True,
+)
+
+out = tokenizer.decode(tokens[0][input_ids.shape[1]:], skip_special_tokens=True).strip()
+print(out)
+```
+
+## 謝辞
+元モデルを開発してくれていた人をはじめ、今まで高い志をもって開発してくれたすべての人に心より感謝を申し上げます。