Format code & move functions (#155)

This commit is contained in:
Lianmin Zheng
2024-02-06 13:27:46 -08:00
committed by GitHub
parent a7334aeea1
commit 23f05005fd
14 changed files with 94 additions and 54 deletions

View File

@@ -12,6 +12,7 @@ import argparse
import requests
def test_decode(url, return_logprob):
response = requests.post(
url + "/generate",
@@ -27,6 +28,7 @@ def test_decode(url, return_logprob):
)
print(response.json())
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--host", type=str, default="http://127.0.0.1")