Update README.md

README.md

@@ -130,7 +130,7 @@ pip install accelerate
 ```
 
 Use the following code to get the quantized model:
-```
+```Py
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TorchAoConfig