mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-11-05 18:32:14 +08:00
15 lines
413 B
Python
15 lines
413 B
Python
import os
|
|
|
|
from llmtuner.webui.interface import create_ui
|
|
|
|
|
|
def main():
|
|
server_name = os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0")
|
|
server_port = int(os.environ.get("GRADIO_SERVER_PORT", "7860"))
|
|
gradio_share = bool(int(os.environ.get("GRADIO_SHARE", "0")))
|
|
create_ui().queue().launch(share=gradio_share, server_name=server_name, server_port=server_port)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|