From fad36d9c0874be58e4d326ecf9f89d143c18382b Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Sat, 1 Apr 2023 16:42:06 +0530 Subject: [PATCH] Don't show or allow test_diffusers if not using the beta branch. This allows allow features to release to the main branch, while restricting diffusers to the beta branch --- ui/easydiffusion/renderer.py | 4 +++- ui/media/js/parameters.js | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/ui/easydiffusion/renderer.py b/ui/easydiffusion/renderer.py index 8270d232..c8004fee 100644 --- a/ui/easydiffusion/renderer.py +++ b/ui/easydiffusion/renderer.py @@ -29,7 +29,9 @@ def init(device): from easydiffusion import app app_config = app.getConfig() - context.test_diffusers = app_config.get("test_diffusers", False) + context.test_diffusers = ( + app_config.get("test_diffusers", False) and app_config.get("update_branch", "main") != "main" + ) device_manager.device_init(context, device) diff --git a/ui/media/js/parameters.js b/ui/media/js/parameters.js index baa55469..eb79a2f0 100644 --- a/ui/media/js/parameters.js +++ b/ui/media/js/parameters.js @@ -309,6 +309,8 @@ async function getAppConfig() { if (config.update_branch === 'beta') { useBetaChannelField.checked = true document.querySelector("#updateBranchLabel").innerText = "(beta)" + } else { + getParameterSettingsEntry("test_diffusers").style.display = "none" } if (config.ui && config.ui.open_browser_on_start === false) { uiOpenBrowserOnStartField.checked = false @@ -320,7 +322,7 @@ async function getAppConfig() { listenPortField.value = config.net.listen_port } if (config.test_diffusers !== undefined) { - testDiffusers.checked = config.test_diffusers + testDiffusers.checked = config.test_diffusers && config.update_branch === 'beta' document.querySelector("#lora_model_container").style.display = (testDiffusers.checked ? '' : 'none') }