PeterZentai
commited on
Commit
•
da1156e
1
Parent(s):
1265ee3
Update README.md
Browse files
README.md
CHANGED
@@ -57,7 +57,7 @@ payload = tokenizer.apply_chat_template([
|
|
57 |
{ 'role': 'assistant', 'content': '''Roger that.''' },
|
58 |
{ 'role': 'user', 'content': '''Bananas slot should be 20''' },
|
59 |
{ 'role': 'assistant', 'content': '''Certainly''' },
|
60 |
-
{ 'role': 'user', 'content': '''What is value of
|
61 |
], tokenize=False, add_generation_prompt=True,)
|
62 |
device = "cuda"
|
63 |
model_input = tokenizer(payload, return_tensors="pt").to(device)
|
|
|
57 |
{ 'role': 'assistant', 'content': '''Roger that.''' },
|
58 |
{ 'role': 'user', 'content': '''Bananas slot should be 20''' },
|
59 |
{ 'role': 'assistant', 'content': '''Certainly''' },
|
60 |
+
{ 'role': 'user', 'content': '''What is the value of Apple + Banana?''' },
|
61 |
], tokenize=False, add_generation_prompt=True,)
|
62 |
device = "cuda"
|
63 |
model_input = tokenizer(payload, return_tensors="pt").to(device)
|