Update README.md

README.md

@@ -46,7 +46,7 @@ model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
 inputs = tokenizer.encode("def print_hello_world():", return_tensors="pt").to(device)
 outputs = model.generate(inputs)
 print(tokenizer.decode(outputs[0]))
-
+```
 * _Using `torch.bfloat16`_
 ```python
 # pip install accelerate