Update README.md

README.md

@@ -31,8 +31,10 @@ The additional details of the Aquila model will be presented in the official tec
 ### 1. Inference
 
 ```python
-from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import BitsAndBytesConfig
+
 device = torch.device("cuda")
 model_info = "BAAI/Aquila2-34B"
 tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)