Update README.md

README.md

@@ -7,6 +7,7 @@ base_model:
 
 Bitsandbytes quantization of https://huggingface.co/bigcode/starcoder2-3b. See https://huggingface.co/blog/4bit-transformers-bitsandbytes for instructions.
 
+```python
 from transformers import AutoModelForCausalLM
 from transformers import BitsAndBytesConfig
 import torch
@@ -18,4 +19,5 @@ nf4_config = BitsAndBytesConfig(
    bnb_4bit_compute_dtype=torch.bfloat16
 )
 model = AutoModelForCausalLM.from_pretrained("bigcode/starcoder2-3b", quantization_config=nf4_config)
-model.push_to_hub("onekq-ai/starcoder2-3b-bnb-4bit")
+model.push_to_hub("onekq-ai/starcoder2-3b-bnb-4bit")
+```