Former-commit-id: c36c5c61fc022b3f144d4c798ec584c4954b0181
This commit is contained in:
hiyouga
2024-10-29 13:02:13 +00:00
parent 22912eba1a
commit 43bd1b070c
5 changed files with 16 additions and 14 deletions

View File

@@ -18,8 +18,9 @@ from llamafactory.webui.interface import create_ui
def main():
gradio_share = os.environ.get("GRADIO_SHARE", "0").lower() in ["true", "1"]
server_name = os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0")
gradio_ipv6 = os.getenv("GRADIO_IPV6", "0").lower() in ["true", "1"]
gradio_share = os.getenv("GRADIO_SHARE", "0").lower() in ["true", "1"]
server_name = os.getenv("GRADIO_SERVER_NAME", "[::]" if gradio_ipv6 else "0.0.0.0")
create_ui().queue().launch(share=gradio_share, server_name=server_name, inbrowser=True)