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

fix gradio args

Browse Source 
Former-commit-id: a153039380d8aa2cdbf434f71f304b1c53ce09f2
This commit is contained in:
hiyouga 2024-05-06 23:33:06 +08:00
parent f2481e78af
commit 89e7cabaa9
2 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/llmtuner/webui/interface.py
View File

@ -1,3 +1,5 @@
import os
from ..extras.packages import is_gradio_available from ..extras.packages import is_gradio_available
from .common import save_config from .common import save_config
from .components import ( from .components import (
@ -69,8 +71,14 @@ def create_web_demo() -> gr.Blocks:
def run_web_ui() -> None: def run_web_ui() -> None:
create_ui().queue().launch() 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)
def run_web_demo() -> None: def run_web_demo() -> None:
create_web_demo().queue().launch() 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_web_demo().queue().launch(share=gradio_share, server_name=server_name, server_port=server_port)

7
src/webui.py
View File

@ -1,8 +1,13 @@
import os
from llmtuner.webui.interface import create_ui from llmtuner.webui.interface import create_ui
def main(): def main():
create_ui().queue().launch(server_name="0.0.0.0", server_port=None, share=False) 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__": if __name__ == "__main__":