This commit is contained in:
hiyouga committed 2024-05-07 17:50:27 +08:00
1 parent 6159acbaa0
commit b0888262e3
7 files changed
+43 -5

No files matched your search

+19
View File
@@ -0,0 +1,19 @@
import os
import uvicorn
from llmtuner.api.app import create_app
from llmtuner.chat import ChatModel
def main():
chat_model = ChatModel()
app = create_app(chat_model)
api_host = os.environ.get("API_HOST", "0.0.0.0")
api_port = int(os.environ.get("API_PORT", "8000"))
print("Visit http://localhost:{}/docs for API document.".format(api_port))
uvicorn.run(app, host=api_host, port=api_port)
if __name__ == "__main__":
main()