423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-03 20:22:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

tiny fix

Browse Source 
Former-commit-id: 0a9986160ccdc9c82efbc83f36d1a3d84859f098
This commit is contained in:
hiyouga 2024-01-07 17:17:18 +08:00
parent 5b61063048
commit db545efd6d
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__":