Update README.md

README.md

@@ -19,10 +19,22 @@ This project is supported by [GENIAC](https://www.meti.go.jp/policy/mono_info_se
 
 ```python
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoTokenizer
+from peft import AutoPeftModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("stockmark/stockmark-100b")
-model = AutoModelForCausalLM.from_pretrained("stockmark/stockmark-100b", device_map="auto", torch_dtype=torch.bfloat16)
+model = AutoPeftModelForCausalLM.from_pretrained("stockmark/stockmark-100b", device_map="auto", torch_dtype=torch.bfloat16)
+
+prompt_template = """### 指示:
+{instruction}
+
+### 応答:
+"""
+
+tokenizer = AutoTokenizer.from_pretrained("stockmark/stockmark-100b-instruct-v0.1")
+model = AutoPeftModelForCausalLM.from_pretrained("stockmark/stockmark-100b-instruct-v0.1", device_map="auto", torch_dtype=torch.bfloat16)
+#
+#
 
 input_ids = tokenizer("生成AIとは？", return_tensors="pt").input_ids.to(model.device)
 with torch.inference_mode():