diff --git a/modules/ui.py b/modules/ui.py index 4a54610d92c..b43b3e2e048 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -615,6 +615,24 @@ def refresh_sd_models(request: gr.Request): return gr.update(**(args or {})) + def refresh_sd_models(request: gr.Request): + tokens = shared.demo.server_app.tokens + cookies = request.headers['cookie'].split('; ') + access_token = None + for cookie in cookies: + if cookie.startswith('access-token'): + access_token = cookie[len('access-token=') : ] + break + username = tokens[access_token] if access_token else None + + refresh_method(username) + args = refreshed_args() if callable(refreshed_args) else refreshed_args + + for k, v in args.items(): + setattr(refresh_component, k, v) + + return gr.update(**(args or {})) + def refresh_checkpoints(sagemaker_endpoint): refresh_method(sagemaker_endpoint) args = refreshed_args() if callable(refreshed_args) else refreshed_args @@ -1579,7 +1597,6 @@ def update_orig(image, state): fn=modules.extras.clear_cache, inputs=[], outputs=[] ) - with gr.Blocks(analytics_enabled=False) as modelmerger_interface: with gr.Row().style(equal_height=False): with gr.Column(variant='panel'):