Update README.md

README.md

@@ -15,6 +15,30 @@ DPOには[Low-Rank Adaptation (LoRA)](https://huggingface.co/docs/peft/conceptua
 ## Requirements, Usage, Chat Template
 
 [cyberagent/calm2-7b-chat](https://huggingface.co/cyberagent/calm2-7b-chat)と同様です。
+同様のコード・プロンプトで動かすことができます。
+
+```python
+import transformers
+from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
+
+assert transformers.__version__ >= "4.34.1"
+
+model = AutoModelForCausalLM.from_pretrained("cyberagent/calm2-7b-chat-dpo-experimental", device_map="auto", torch_dtype="auto")
+tokenizer = AutoTokenizer.from_pretrained("cyberagent/calm2-7b-chat-dpo-experimental")
+streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
+
+prompt = """USER: AIによって私達の暮らしはどのように変わりますか？
+ASSISTANT: """
+
+token_ids = tokenizer.encode(prompt, return_tensors="pt")
+output_ids = model.generate(
+    input_ids=token_ids.to(model.device),
+    max_new_tokens=300,
+    do_sample=True,
+    temperature=0.8,
+    streamer=streamer,
+)
+```
 
 ## 実験結果
 
@@ -46,6 +70,9 @@ DPOには[Low-Rank Adaptation (LoRA)](https://huggingface.co/docs/peft/conceptua
 | stem	| 6.3	| 6.2 |
 | writing	| 7.7	| 9.1 |
 
+## Releases
+
+1.0: v1 release (Jan 24, 2024)
 
 ## Author