Update README.md

README.md

@@ -53,6 +53,7 @@ XML likeなinstructionテンプレートを採用しました
 
 ```
 
+Example
 ```
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
@@ -62,6 +63,9 @@ model_name = "aixsatoshi/Honyaku-Multi-Translator-Swallow-ms7b"
 model = AutoModelForCausalLM.from_pretrained(model_name).to(device)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
+# Move model to the appropriate device and set to bf16 precision
+model.to(device).to(dtype=torch.bfloat16)
+
 # Define the English prompt
 english_prompt = "What is your favourite condiment?"