PeterZentai
commited on
Commit
•
9a9f518
1
Parent(s):
da1156e
Update README.md
Browse files
README.md
CHANGED
@@ -52,7 +52,7 @@ model = AutoModelForCausalLM.from_pretrained(model_id, device_map={"": 0}, trust
|
|
52 |
tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=False, trust_remote_code=True)
|
53 |
|
54 |
payload = tokenizer.apply_chat_template([
|
55 |
-
{ 'role': 'system', 'content': '''You are a state machine. The user will add state slot values and
|
56 |
{ 'role': 'user', 'content': '''Place 15 into slot apple''' },
|
57 |
{ 'role': 'assistant', 'content': '''Roger that.''' },
|
58 |
{ 'role': 'user', 'content': '''Bananas slot should be 20''' },
|
|
|
52 |
tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=False, trust_remote_code=True)
|
53 |
|
54 |
payload = tokenizer.apply_chat_template([
|
55 |
+
{ 'role': 'system', 'content': '''You are a state machine. The user will add state slot values and you'll keep track of them.''' },
|
56 |
{ 'role': 'user', 'content': '''Place 15 into slot apple''' },
|
57 |
{ 'role': 'assistant', 'content': '''Roger that.''' },
|
58 |
{ 'role': 'user', 'content': '''Bananas slot should be 20''' },
|