diff --git a/ui/media/js/parameters.js b/ui/media/js/parameters.js
index 022b9a7f..455844da 100644
--- a/ui/media/js/parameters.js
+++ b/ui/media/js/parameters.js
@@ -321,11 +321,11 @@ async function getAppConfig() {
         if (config.net && config.net.listen_port !== undefined) {
             listenPortField.value = config.net.listen_port
         }
-        if (config.test_diffusers === undefined) {
+        if (config.test_diffusers === undefined || config.update_branch === 'main') {
             document.querySelector("#lora_model_container").style.display = 'none'
             document.querySelector("#lora_alpha_container").style.display = 'none'
         } else {
-            testDiffusers.checked = config.test_diffusers && config.update_branch === 'beta'
+            testDiffusers.checked = config.test_diffusers && config.update_branch !== 'main'
             document.querySelector("#lora_model_container").style.display = (testDiffusers.checked ? '' : 'none')
             document.querySelector("#lora_alpha_container").style.display = (testDiffusers.checked && loraModelField.value !== "" ? '' : 'none')
         }