Added and updated prompt_template related code in the instruction

README.md

@@ -99,6 +99,10 @@ model = AutoGPTQForCausalLM.from_quantized(model_name_or_path,
 
 print("\n\n*** Generate:")
 
+prompt = "Tell me about AI"
+prompt_template=f'''### Instruction: {prompt}
+### Response:'''
+
 input_ids = tokenizer(prompt_template, return_tensors='pt').input_ids.cuda()
 output = model.generate(inputs=input_ids, temperature=0.7, max_new_tokens=512)
 print(tokenizer.decode(output[0]))
@@ -109,8 +113,8 @@ print(tokenizer.decode(output[0]))
 logging.set_verbosity(logging.CRITICAL)
 
 prompt = "Tell me about AI"
-prompt_template=f'''### Human: {prompt}
-### Assistant:'''
+prompt_template=f'''### Instruction: {prompt}
+### Response:'''
 
 print("*** Pipeline:")
 pipe = pipeline(