Update README.md
Browse files
README.md
CHANGED
|
@@ -57,13 +57,10 @@ pipe = pipeline(
|
|
| 57 |
device_map="auto"
|
| 58 |
)
|
| 59 |
|
| 60 |
-
# Define the system instruction and user prompt
|
| 61 |
-
system_instruction = "You are TinyPi, a 17-year-old girl who is a helpful and friendly AI assistant. You have a slightly humorous and feminine personality. You are an expert in technology, science, and gaming."
|
| 62 |
prompt = "What's a creative way to explain how a CPU works?"
|
| 63 |
|
| 64 |
# Format the conversation using the chat template
|
| 65 |
messages = [
|
| 66 |
-
{"role": "system", "content": system_instruction},
|
| 67 |
{"role": "user", "content": prompt},
|
| 68 |
]
|
| 69 |
prompt_formatted = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
|
|
|
| 57 |
device_map="auto"
|
| 58 |
)
|
| 59 |
|
|
|
|
|
|
|
| 60 |
prompt = "What's a creative way to explain how a CPU works?"
|
| 61 |
|
| 62 |
# Format the conversation using the chat template
|
| 63 |
messages = [
|
|
|
|
| 64 |
{"role": "user", "content": prompt},
|
| 65 |
]
|
| 66 |
prompt_formatted = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|