p04798526
/
LLaMA-Factory-Mirror
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tiny fix

This commit is contained in:
hiyouga 2024-01-07 17:17:18 +08:00
parent 08464183b9
commit 0a9986160c
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/api_demo.py
View File

@ -1,3 +1,4 @@
import os
import uvicorn import uvicorn
from llmtuner import ChatModel, create_app from llmtuner import ChatModel, create_app
@ -6,8 +7,8 @@ from llmtuner import ChatModel, create_app
def main(): def main():
chat_model = ChatModel() chat_model = ChatModel()
app = create_app(chat_model) app = create_app(chat_model)
print("Visit http://localhost:8000/docs for API document.") print("Visit http://localhost:{}/docs for API document.".format(os.environ.get("API_PORT", 8000)))
uvicorn.run(app, host="0.0.0.0", port=8000, workers=1) uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("API_PORT", 8000)), workers=1)
if __name__ == "__main__": if __name__ == "__main__":