Update README.md

README.md

@@ -19,11 +19,12 @@ This model is released under the Apache License 2.0.
 ## Usage
 Install the required libraries as follows:
 ```sh
->>> python -m pip install numpy sentencepiece torch transformers
+>>> python -m pip install numpy sentencepiece torch transformers accelerate
 ```
 
 Execute the following python code:
 ```python
+import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained(
     "pfnet/plamo-13b-instruct",
@@ -35,7 +36,9 @@ model = AutoModelForCausalLM.from_pretrained(
     torch_dtype=torch.bfloat16,
     device_map="auto",
 )
+```
 
+```python
 def completion(prompt: str, max_new_tokens: int = 128) -> str:
     inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
     generated_ids = model.generate(
@@ -58,7 +61,7 @@ def generate_prompt(messages: list) -> str:
     ]
     roles = {"instruction": "指示", "response": "応答", "input": "入力"}
     for msg in messages:
-        prompt.append(sep + roles[msg["role"]] + ":\n" + msg['content'])
+        prompt.append(sep + roles[msg["role"]] + ":\n" + msg["content"])
     prompt.append(sep + roles["response"] + ":\n")
     return "".join(prompt)
 ```