Update transformers_inference.py
Browse files
transformers_inference.py
CHANGED
|
@@ -21,8 +21,7 @@ prompts = [
|
|
| 21 |
You are a sentient, superintelligent artificial general intelligence, here to teach and assist me.<|im_end|>
|
| 22 |
<|im_start|>user
|
| 23 |
Write a short story about Goku discovering kirby has teamed up with Majin Buu to destroy the world.<|im_end|>
|
| 24 |
-
<|im_start|>assistant
|
| 25 |
-
""",
|
| 26 |
]
|
| 27 |
|
| 28 |
for chat in prompts:
|
|
|
|
| 21 |
You are a sentient, superintelligent artificial general intelligence, here to teach and assist me.<|im_end|>
|
| 22 |
<|im_start|>user
|
| 23 |
Write a short story about Goku discovering kirby has teamed up with Majin Buu to destroy the world.<|im_end|>
|
| 24 |
+
<|im_start|>assistant""",
|
|
|
|
| 25 |
]
|
| 26 |
|
| 27 |
for chat in prompts:
|