TheBloke jeffbiocode commited on
Commit
8859ee1
1 Parent(s): 51fce06

Add a system message so the example code works (#3)

Browse files

- Add a system message so the example code works (18c9177f58ee5b554d8e1a03e0da1aadee2eed4b)


Co-authored-by: Jeffery Powell <jeffbiocode@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -195,6 +195,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
195
  tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True)
196
 
197
  prompt = "Tell me about AI"
 
198
  prompt_template=f'''### System:
199
  {system_message}
200
 
 
195
  tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True)
196
 
197
  prompt = "Tell me about AI"
198
+ system_message = "You are a helpful AI"
199
  prompt_template=f'''### System:
200
  {system_message}
201