Update README.md

README.md

@@ -19,12 +19,13 @@ This model is released under CC-BY-NC-4.0.
 ## Usage
 Install the required libraries as follows:
 ```sh
->>> python -m pip install numpy safetensors sentencepiece torch transformers
+>>> python -m pip install numpy safetensors 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-nc",
@@ -36,7 +37,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(