upstream-archive/textgen
1
0
Fork 0
mirror of https://github.com/oobabooga/textgen.git synced 2026-07-23 11:20:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

Electron: Refresh models dropdown when a new models directory is selected

Browse Source
This commit is contained in:
oobabooga
2026-05-17 21:54:20 -07:00
parent 2b07e17c52
commit fa57f838b2
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modules/ui_session.py
View File

@@ -109,7 +109,7 @@ def create_ui():
None, gradio('model_dir'), gradio('model_dir'),
js='async (current) => { const p = await window.electronAPI.pickDirectory(); return p === null ? current : p; }')
shared.gradio['model_dir'].change(apply_model_dir, gradio('model_dir'), None, show_progress=False)
shared.gradio['model_dir'].change(apply_model_dir, gradio('model_dir'), gradio('model_menu', 'model_draft'), show_progress=False)
shared.gradio['show_two_notebook_columns'].change(
handle_default_to_notebook_change,
@@ -165,6 +165,10 @@ def apply_model_dir(value):
if Path(value).is_dir():
shared.args.model_dir = value
shared.user_config = shared.load_user_config()
models = utils.get_available_models()
return gr.update(choices=models), gr.update(choices=['None'] + models)
return gr.update(), gr.update()
def set_interface_arguments(extensions, bool_active):