Former-commit-id: 0a9986160ccdc9c82efbc83f36d1a3d84859f098
This commit is contained in:
hiyouga 2024-01-07 17:17:18 +08:00
parent 5b61063048
commit db545efd6d

View File

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