add preset refresh on "Refresh All Files" click

This commit is contained in:
Manuel Schmid 2023-12-23 22:10:50 +01:00
parent 2770a40dc1
commit ccfd4f5094
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
2 changed files with 8 additions and 2 deletions

View File

@ -90,6 +90,9 @@ def list_presets():
available_presets = list_presets()
def update_presets():
global available_presets
available_presets = list_presets()
def try_get_preset_content(preset):
if isinstance(preset, str):

View File

@ -451,13 +451,16 @@ with shared.gradio_root:
def model_refresh_clicked():
modules.config.update_all_model_names()
modules.config.update_presets()
results = []
results += [gr.update(choices=modules.config.model_filenames), gr.update(choices=['None'] + modules.config.model_filenames)]
results += [gr.update(choices=modules.config.model_filenames),
gr.update(choices=['None'] + modules.config.model_filenames),
gr.update(choices=modules.config.available_presets)]
for i in range(5):
results += [gr.update(choices=['None'] + modules.config.lora_filenames), gr.update()]
return results
model_refresh.click(model_refresh_clicked, [], [base_model, refiner_model] + lora_ctrls,
model_refresh.click(model_refresh_clicked, [], [base_model, refiner_model, preset_selection] + lora_ctrls,
queue=False, show_progress=False)
def preset_selection_change(preset):