diff options
| author | Rickard Edén <rickardeden@gmail.com> | 2023-12-03 10:03:25 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-03 11:03:25 +0200 |
| commit | 6949b50df56ee58a2d76d45487942cb211c08629 (patch) | |
| tree | d01555833ffd671955bc6a308783d7c6874dfd98 /examples/server/api_like_OAI.py | |
| parent | d7b800b8bc490a221acbd83c575206a907f2f6e2 (diff) | |
py : add grammar to oai like api (#4294)
Diffstat (limited to 'examples/server/api_like_OAI.py')
| -rwxr-xr-x | examples/server/api_like_OAI.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/examples/server/api_like_OAI.py b/examples/server/api_like_OAI.py index 830c056d..607fe49d 100755 --- a/examples/server/api_like_OAI.py +++ b/examples/server/api_like_OAI.py @@ -70,6 +70,7 @@ def make_postData(body, chat=False, stream=False): if(is_present(body, "mirostat_tau")): postData["mirostat_tau"] = body["mirostat_tau"] if(is_present(body, "mirostat_eta")): postData["mirostat_eta"] = body["mirostat_eta"] if(is_present(body, "seed")): postData["seed"] = body["seed"] + if(is_present(body, "grammar")): postData["grammar"] = body["grammar"] if(is_present(body, "logit_bias")): postData["logit_bias"] = [[int(token), body["logit_bias"][token]] for token in body["logit_bias"].keys()] if (args.stop != ""): postData["stop"] = [args.stop] |