Update README.md

README.md

@@ -10,7 +10,7 @@ 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 AutoModelForCausalLM, AutoTokenizer
 from transformers import BitsAndBytesConfig
 import torch
 
@@ -21,5 +21,8 @@ nf4_config = BitsAndBytesConfig(
  bnb_4bit_compute_dtype=torch.bfloat16
 )
 model = AutoModelForCausalLM.from_pretrained("bigcode/starcoder2-3b", quantization_config=nf4_config)
+tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoder2-3b")
+
 model.push_to_hub("onekq-ai/starcoder2-3b-bnb-4bit")
+tokenizer.push_to_hub("onekq-ai/starcoder2-3b-bnb-4bit")
 ```