Compare commits
3 Commits
aa758e3c2a
...
d0aebd17fa
Author | SHA1 | Date | |
---|---|---|---|
![]() |
d0aebd17fa | ||
![]() |
d9abf08184 | ||
![]() |
a27a1ab079 |
@ -61,9 +61,11 @@ def prompt_manage_page():
|
||||
|
||||
selected_row = None # 保存当前选中行的全局变量
|
||||
|
||||
def select_record(evt: gr.SelectData):
|
||||
def select_record(dataFrame ,evt: gr.SelectData):
|
||||
global selected_row
|
||||
selected_row = evt.row_value
|
||||
selected_row = dataFrame.iloc[evt.index[0]].tolist()
|
||||
selected_row[0] = int(selected_row[0])
|
||||
print(selected_row)
|
||||
|
||||
with gr.Blocks() as demo:
|
||||
gr.Markdown("## 提示词模板管理")
|
||||
@ -102,7 +104,10 @@ def prompt_manage_page():
|
||||
outputs=[prompt_table, name_input, description_input, content_input]
|
||||
)
|
||||
|
||||
prompt_table.select(select_record, [], [], show_progress="hidden")
|
||||
prompt_table.select(fn=select_record,
|
||||
inputs=[prompt_table],
|
||||
outputs=[],
|
||||
show_progress="hidden")
|
||||
|
||||
edit_button.click(
|
||||
fn=edit_prompt,
|
||||
|
@ -68,9 +68,11 @@ def setting_page():
|
||||
|
||||
selected_row = None # 保存当前选中行的全局变量
|
||||
|
||||
def select_record(evt: gr.SelectData):
|
||||
def select_record(dataFrame ,evt: gr.SelectData):
|
||||
global selected_row
|
||||
selected_row = evt.row_value
|
||||
selected_row = dataFrame.iloc[evt.index[0]].tolist()
|
||||
selected_row[0] = int(selected_row[0])
|
||||
print(selected_row)
|
||||
|
||||
with gr.Blocks() as demo:
|
||||
gr.Markdown("## API Provider 管理")
|
||||
@ -109,7 +111,10 @@ def setting_page():
|
||||
outputs=[provider_table, model_id_input, base_url_input, api_key_input] # 添加清空输入框的输出
|
||||
)
|
||||
|
||||
provider_table.select(select_record, [], [], show_progress="hidden")
|
||||
provider_table.select(fn=select_record,
|
||||
inputs=[provider_table],
|
||||
outputs=[],
|
||||
show_progress="hidden")
|
||||
|
||||
edit_button.click(
|
||||
fn=edit_provider,
|
||||
|
@ -16,17 +16,15 @@ def train_page():
|
||||
# 获取数据集列表并设置初始值
|
||||
datasets_list = [str(ds["name"]) for ds in get_datasets().all()]
|
||||
initial_dataset = datasets_list[0] if datasets_list else None
|
||||
|
||||
dataset_dropdown = gr.Dropdown(
|
||||
choices=datasets_list,
|
||||
value=initial_dataset, # 设置初始选中项
|
||||
label="选择数据集",
|
||||
allow_custom_value=True,
|
||||
interactive=True
|
||||
)
|
||||
|
||||
with gr.Row():
|
||||
with gr.Column(scale=1):
|
||||
dataset_dropdown = gr.Dropdown(
|
||||
choices=datasets_list,
|
||||
value=initial_dataset, # 设置初始选中项
|
||||
label="选择数据集",
|
||||
allow_custom_value=True,
|
||||
interactive=True
|
||||
)
|
||||
# 新增超参数输入组件
|
||||
learning_rate_input = gr.Number(value=2e-4, label="学习率")
|
||||
per_device_train_batch_size_input = gr.Number(value=1, label="batch size", precision=0)
|
||||
@ -37,7 +35,7 @@ def train_page():
|
||||
train_button = gr.Button("开始微调")
|
||||
|
||||
# 训练状态输出
|
||||
output = gr.Textbox(label="训练日志", interactive=False)
|
||||
output = gr.Textbox(label="训练状态", interactive=False)
|
||||
with gr.Column(scale=3):
|
||||
# 新增 TensorBoard iframe 显示框
|
||||
tensorboard_iframe = gr.HTML(label="TensorBoard 可视化")
|
||||
|
@ -3,16 +3,14 @@ from tools import scan_docs_directory
|
||||
|
||||
_prompt_store = None
|
||||
_sql_engine = None
|
||||
_docs = None
|
||||
_datasets = None
|
||||
_model = None
|
||||
_tokenizer = None
|
||||
_workdir = None
|
||||
def init_global_var(workdir="workdir"):
|
||||
global _prompt_store, _sql_engine, _docs, _datasets, _workdir
|
||||
global _prompt_store, _sql_engine, _datasets, _workdir
|
||||
_prompt_store = get_prompt_tinydb(workdir)
|
||||
_sql_engine = get_sqlite_engine(workdir)
|
||||
_docs = scan_docs_directory(workdir)
|
||||
_datasets = get_all_dataset(workdir)
|
||||
_workdir = workdir
|
||||
|
||||
@ -34,12 +32,8 @@ def set_sql_engine(new_sql_engine):
|
||||
_sql_engine = new_sql_engine
|
||||
|
||||
def get_docs():
|
||||
return _docs
|
||||
|
||||
def set_docs(new_docs):
|
||||
global _docs
|
||||
_docs = new_docs
|
||||
|
||||
global _workdir
|
||||
return scan_docs_directory(_workdir)
|
||||
def get_datasets():
|
||||
return _datasets
|
||||
|
||||
|
@ -1,7 +1,7 @@
|
||||
openai>=1.0.0
|
||||
python-dotenv>=1.0.0
|
||||
pydantic>=2.0.0
|
||||
gradio>=5.0.0
|
||||
gradio>=5.25.0
|
||||
langchain>=0.3
|
||||
tinydb>=4.0.0
|
||||
unsloth>=2025.3.19
|
||||
|
Loading…
x
Reference in New Issue
Block a user