Update README.md

README.md

@@ -29,6 +29,15 @@ 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/WizardLM-7B-V1.0-Uncensored-GGML)
 * [Unquantised fp16 model in pytorch format, for GPU inference and for further conversions](https://huggingface.co/ehartford/WizardLM-7B-V1.0-Uncensored)
 
+## Prompt template
+
+```
+You are a helpful AI assistant.
+
+USER: <prompt>
+ASSISTANT:
+```
+
 ## 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
@@ -74,8 +83,8 @@ model = AutoGPTQForCausalLM.from_quantized(model_name_or_path,
 
 # Note: check the prompt template is correct for this model.
 prompt = "Tell me about AI"
-prompt_template=f'''### Human: {prompt}
-### Assistant:'''
+prompt_template=f'''USER: {prompt}
+ASSISTANT:'''
 
 print("\n\n*** Generate:")