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') }