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

[U] Compute hostname differently

Browse Source 
Former-commit-id: 6554cdeedb8ab669ce500be404dab8d42a90779b
This commit is contained in:
Azalea 2024-11-28 22:23:41 -05:00 committed by GitHub
parent f5e6e25a1b
commit 0efa34c9ef
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
import os import os
import socket import platform
from ..extras.packages import is_gradio_available from ..extras.packages import is_gradio_available
from .common import save_config from .common import save_config
@ -35,8 +35,9 @@ if is_gradio_available():
def create_ui(demo_mode: bool = False) -> "gr.Blocks": def create_ui(demo_mode: bool = False) -> "gr.Blocks":
engine = Engine(demo_mode=demo_mode, pure_chat=False) engine = Engine(demo_mode=demo_mode, pure_chat=False)
hostname = os.getenv("HOSTNAME", os.getenv("COMPUTERNAME", platform.node())).split('.')[0]
with gr.Blocks(title=f"LLaMA Board ({socket.gethostname()})", css=CSS) as demo: with gr.Blocks(title=f"LLaMA Board ({hostname})", css=CSS) as demo:
if demo_mode: if demo_mode:
gr.HTML("<h1><center>LLaMA Board: A One-stop Web UI for Getting Started with LLaMA Factory</center></h1>") gr.HTML("<h1><center>LLaMA Board: A One-stop Web UI for Getting Started with LLaMA Factory</center></h1>")
gr.HTML( gr.HTML(