reduce tokens for backend tester
This commit is contained in:
parent
9235725bdd
commit
bc25d92c95
|
@ -12,7 +12,7 @@ def test_backend(backend_url: str, test_prompt: bool = False):
|
||||||
"prompt": "Test prompt",
|
"prompt": "Test prompt",
|
||||||
"stream": False,
|
"stream": False,
|
||||||
"temperature": 0,
|
"temperature": 0,
|
||||||
"max_new_tokens": 16,
|
"max_new_tokens": 3,
|
||||||
}
|
}
|
||||||
success, response, err = generator(data, backend_url, timeout=10)
|
success, response, err = generator(data, backend_url, timeout=10)
|
||||||
if not success or not response or err:
|
if not success or not response or err:
|
||||||
|
|
Reference in New Issue