tests: make sure all prompts are the same

This commit is contained in:
Cyberes 2023-10-20 17:35:21 -06:00
parent 763139c949
commit 1a15232400
3 changed files with 3 additions and 3 deletions

View File

@ -33,7 +33,7 @@ while true; do
DATA=$(
cat <<EOF
{
"prompt": "Please give me a step-by-step guide on how to plant a tree in my backyard.",
"prompt": "Write a 300 word story about an apple tree.",
"temperature": 1,
"max_new_tokens": 100,
"top_p": 1.0,

View File

@ -36,7 +36,7 @@ while true; do
cat <<EOF
{
"model": "gpt-4",
"prompt": "Please give me a step-by-step guide on how to plant a tree in my backyard.",
"prompt": "Write a 300 word story about an apple tree.",
"max_tokens": 100,
"stream": $DO_STREAM
}

View File

@ -98,7 +98,7 @@ async def print_response_stream(prompt):
if __name__ == '__main__':
prompt = "Write a 300 word description of how an apple tree grows.\n\n"
prompt = "Write a 300 word story about an apple tree.\n\n"
while True:
print('--> START <--')
asyncio.run(print_response_stream(prompt))