Update README.md

README.md

@@ -76,14 +76,14 @@ bnb_config = BitsAndBytesConfig(
     bnb_4bit_compute_dtype=getattr(torch, "float16"),
     bnb_4bit_use_double_quant=False)
 model = AutoModelForCausalLM.from_pretrained(
-        "meta-llama/Llama-2-13b-hf",
+        "meta-llama/Llama-2-13b-chat-hf",
         quantization_config=bnb_config,
         device_map={"": 0})
 model.config.use_cache = False
 model.config.pretraining_tp = 1
 model = PeftModel.from_pretrained(model, "TuningAI/Llama2_13B_startup_Assistant")
 
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-13b-hf", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-13b-chat-hf", trust_remote_code=True)
 tokenizer.pad_token = tokenizer.eos_token
 tokenizer.padding_side = "right"
 system_message = "Given a user's startup-related question in English, you will generate a thoughtful answer in English."