Update README.md

README.md

@@ -29,6 +29,13 @@ It is the result of quantising to 4bit using [GPTQ-for-LLaMa](https://github.com
 * [2, 3, 4, 5, 6 and 8-bit GGML models for CPU+GPU inference](https://huggingface.co/TheBloke/h2ogpt-research-oasst1-llama-65B-GGML)
 * [Unquantised fp16 model in pytorch format, for GPU inference and for further conversions](https://huggingface.co/h2oai/h2ogpt-research-oasst1-llama-65b)
 
+## Prompt template
+
+```
+<human>: prompt
+<bot>:
+```
+
 ## How to easily download and use this model in text-generation-webui
 
 Please make sure you're using the latest version of text-generation-webui
@@ -72,10 +79,9 @@ model = AutoGPTQForCausalLM.from_quantized(model_name_or_path,
         use_triton=use_triton,
         quantize_config=None)
 
-# Note: check the prompt template is correct for this model.
 prompt = "Tell me about AI"
-prompt_template=f'''USER: {prompt}
-ASSISTANT:'''
+prompt_template=f'''<human>: {prompt}
+<bot>:'''
 
 print("\n\n*** Generate:")