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