Upload README.md

README.md

@@ -7,7 +7,9 @@ license: mit
 model_creator: Nobody.png
 model_name: Yi 34B GiftedConvo Llama
 model_type: llama
-prompt_template: '{prompt}
+prompt_template: 'USER: {prompt}
+
+  ASSISTANT:
 
   '
 quantized_by: TheBloke
@@ -55,10 +57,11 @@ These files were quantised using hardware kindly provided by [Massed Compute](ht
 <!-- repositories-available end -->
 
 <!-- prompt-template start -->
-## Prompt template: Unknown
+## Prompt template: User-Assistant
 
 ```
-{prompt}
+USER: {prompt}
+ASSISTANT:
 
 ```
 
@@ -236,7 +239,8 @@ from huggingface_hub import InferenceClient
 endpoint_url = "https://your-endpoint-url-here"
 
 prompt = "Tell me about AI"
-prompt_template=f'''{prompt}
+prompt_template=f'''USER: {prompt}
+ASSISTANT:
 '''
 
 client = InferenceClient(endpoint_url)
@@ -289,7 +293,8 @@ model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True)
 
 prompt = "Tell me about AI"
-prompt_template=f'''{prompt}
+prompt_template=f'''USER: {prompt}
+ASSISTANT:
 '''
 
 print("\n\n*** Generate:")