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

[webui] fix launch with proxy (#7332)

Browse Source
This commit is contained in:
taoharry 2025-04-01 15:52:56 +08:00 committed by GitHub
parent 5d1cc863a4
commit 6faa6fb53d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/llamafactory/webui/interface.py
View File

@ -14,6 +14,8 @@
import os import os
import platform import platform
import httpx
from ..extras.misc import is_env_enabled from ..extras.misc import is_env_enabled
from ..extras.packages import is_gradio_available from ..extras.packages import is_gradio_available
@ -88,10 +90,26 @@ def create_web_demo() -> "gr.Blocks":
def run_web_ui() -> None: def run_web_ui() -> None:
os.environ["no_proxy"] = "localhost,127.0.0.1,0.0.0.0"
gradio_ipv6 = is_env_enabled("GRADIO_IPV6") gradio_ipv6 = is_env_enabled("GRADIO_IPV6")
gradio_share = is_env_enabled("GRADIO_SHARE") gradio_share = is_env_enabled("GRADIO_SHARE")
server_name = os.getenv("GRADIO_SERVER_NAME", "[::]" if gradio_ipv6 else "0.0.0.0") 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) httpx.HTTPCORE_OPTS = {"trust_env": False}
try:
demo = create_ui().queue()
demo.launch(
share=gradio_share,
server_name=server_name,
inbrowser=True,
prevent_thread_lock=False,
show_error=True,
quiet=True,
favicon_path=None
)
except Exception as e:
print(f"Error launching web UI: {str(e)}")
raise
def run_web_demo() -> None: def run_web_demo() -> None: