Update README.md

README.md

@@ -66,12 +66,13 @@ pip install --upgrade autoawq autoawq-kernels
 from awq import AutoAWQForCausalLM
 from transformers import AutoTokenizer, TextStreamer
 
-quant_path = "solidrust/WestLake-7B-v2-laser-AWQ"
+model_path = "solidrust/WestLake-7B-v2-laser-AWQ"
+system_message = "Welcome to WestLake. You are here to help users with any questions they may have."
 
 # Load model
-model = AutoAWQForCausalLM.from_quantized(quant_path,
+model = AutoAWQForCausalLM.from_quantized(model_path,
                                           fuse_layers=True)
-tokenizer = AutoTokenizer.from_pretrained(quant_path,
+tokenizer = AutoTokenizer.from_pretrained(model_path,
                                           trust_remote_code=True)
 streamer = TextStreamer(tokenizer,
                         skip_prompt=True,