refactor(frontend): 优化模型管理页面的交互和显示
- 将状态输出从 Textbox 改为 Label 组件,提高用户体验 - 添加 get_model_name 函数以获取模型名称,提高代码复用性 - 更新模型加载、卸载和保存后的状态显示,使信息更加准确 - 优化模型列表刷新功能,确保模型列表实时更新
This commit is contained in:
parent
cbb3a09dd8
commit
0fa2b51a79
@ -7,6 +7,7 @@ import torch
|
||||
|
||||
sys.path.append(str(Path(__file__).resolve().parent.parent))
|
||||
from global_var import model,tokenizer
|
||||
from tools.model import get_model_name
|
||||
|
||||
def model_manage_page():
|
||||
workdir = "workdir" # 假设workdir是当前工作目录下的一个文件夹
|
||||
@ -15,7 +16,7 @@ def model_manage_page():
|
||||
|
||||
with gr.Blocks() as demo:
|
||||
gr.Markdown("## 模型管理")
|
||||
output_text = gr.Textbox(label="当前状态", interactive=False)
|
||||
state_output = gr.Label(label="当前状态",value="当前未加载模型") # 将 Textbox 改为 Label
|
||||
with gr.Row():
|
||||
with gr.Column(scale=3):
|
||||
model_select_dropdown = gr.Dropdown(choices=model_folders, label="选择模型", interactive=True) # 将子文件夹列表添加到Dropdown组件中,并设置为可选
|
||||
@ -45,11 +46,11 @@ def model_manage_page():
|
||||
max_seq_length=max_seq_length,
|
||||
load_in_4bit=load_in_4bit,
|
||||
)
|
||||
return f"模型 {selected_model} 已加载"
|
||||
return f"模型 {get_model_name(model)} 已加载"
|
||||
except Exception as e:
|
||||
return f"加载模型时出错: {str(e)}"
|
||||
|
||||
load_button.click(fn=load_model, inputs=[model_select_dropdown, max_seq_length_input, load_in_4bit_input], outputs=output_text)
|
||||
load_button.click(fn=load_model, inputs=[model_select_dropdown, max_seq_length_input, load_in_4bit_input], outputs=state_output)
|
||||
|
||||
def unload_model():
|
||||
try:
|
||||
@ -68,11 +69,11 @@ def model_manage_page():
|
||||
# 将模型设置为 None
|
||||
model = None
|
||||
|
||||
return "模型已卸载"
|
||||
return "当前未加载模型"
|
||||
except Exception as e:
|
||||
return f"卸载模型时出错: {str(e)}"
|
||||
|
||||
unload_button.click(fn=unload_model, inputs=None, outputs=output_text)
|
||||
unload_button.click(fn=unload_model, inputs=None, outputs=state_output)
|
||||
|
||||
def save_model(save_model_name):
|
||||
try:
|
||||
@ -88,7 +89,7 @@ def model_manage_page():
|
||||
except Exception as e:
|
||||
return f"保存模型时出错: {str(e)}"
|
||||
|
||||
save_button.click(fn=save_model, inputs=save_model_name_input, outputs=output_text)
|
||||
save_button.click(fn=save_model, inputs=save_model_name_input, outputs=state_output)
|
||||
|
||||
def refresh_model_list():
|
||||
try:
|
||||
|
Loading…
x
Reference in New Issue
Block a user