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>
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 |
|