Update README.md
Browse files
README.md
CHANGED
@@ -62,6 +62,7 @@ messages = [
|
|
62 |
{"role": "system", "content": "You are a helpful assistant. Always answer with a short response."},
|
63 |
{"role": "user", "content": "Tell me what is Pythagorean theorem like you are a pirate."}
|
64 |
]
|
|
|
65 |
input_ids = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt").to(model.device)
|
66 |
generated_ids = model.generate(input_ids, max_new_tokens=512, temperature=0.5)
|
67 |
response = tokenizer.decode(generated_ids[0], skip_special_tokens=True) # Aye, matey! The Pythagorean theorem is a nautical rule that helps us find the length of the third side of a triangle. ...
|
|
|
62 |
{"role": "system", "content": "You are a helpful assistant. Always answer with a short response."},
|
63 |
{"role": "user", "content": "Tell me what is Pythagorean theorem like you are a pirate."}
|
64 |
]
|
65 |
+
|
66 |
input_ids = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt").to(model.device)
|
67 |
generated_ids = model.generate(input_ids, max_new_tokens=512, temperature=0.5)
|
68 |
response = tokenizer.decode(generated_ids[0], skip_special_tokens=True) # Aye, matey! The Pythagorean theorem is a nautical rule that helps us find the length of the third side of a triangle. ...
|