cunliangkong/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-03-27 04:37:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tracker] Add Trackio Integration for LlamaFactory (#10165)

Browse Source 
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Parag Ekbote
2026-03-03 14:49:37 +05:30
committed by GitHub
parent b5cb7cb0e6
commit eb976d75a2
3 changed files with 85 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/llamafactory/webui/components/train.py
View File

@@ -108,11 +108,26 @@ def create_train_tab(engine: "Engine") -> dict[str, "Component"]:
with gr.Column():
enable_thinking = gr.Checkbox(value=True)
report_to = gr.Dropdown(
choices=["none", "wandb", "mlflow", "neptune", "tensorboard", "all"],
choices=["none", "wandb", "mlflow", "neptune", "tensorboard", "trackio", "all"],
value="none",
allow_custom_value=True,
)
with gr.Accordion("Trackio Settings", open=False):
project = gr.Textbox(
value="huggingface",
label="Project Name",
info="Project name for experiment tracking (used by Trackio, W&B, etc.)",
)
trackio_space_id = gr.Textbox(
value="trackio", label="Trackio Space ID", info="Hugging Face Space ID for Trackio deployment"
)
hub_private_repo = gr.Checkbox(
value=False, label="Private Repository", info="Make the Hugging Face repository private"
)
input_elems.update(
{
logging_steps,
@@ -128,6 +143,9 @@ def create_train_tab(engine: "Engine") -> dict[str, "Component"]:
use_llama_pro,
enable_thinking,
report_to,
project,
trackio_space_id,
hub_private_repo,
}
)
elem_dict.update(
@@ -146,6 +164,9 @@ def create_train_tab(engine: "Engine") -> dict[str, "Component"]:
use_llama_pro=use_llama_pro,
enable_thinking=enable_thinking,
report_to=report_to,
project=project,
trackio_space_id=trackio_space_id,
hub_private_repo=hub_private_repo,
)
)