Fix for Forge, #258

2024-08-14 16:43:22 +02:00 · 2024-08-14 16:43:22 +02:00 · f353bc1152
parent 19b4b7fc8f
commit f353bc1152
1 changed files with 5 additions and 1 deletions
--- a/javascript/image_browser.js
+++ b/javascript/image_browser.js
@ -12,7 +12,11 @@ onUiLoaded(image_browser_start_it_up)
 async function image_browser_wait_for_webui() { 
    if (image_browser_debug) console.log("image_browser_wait_for_webui:start")
    await image_browser_delay(500)
-    sd_model = gradioApp().getElementById("setting_sd_model_checkpoint")
+    let sd_model = gradioApp().getElementById("setting_sd_model_checkpoint")
+    if (!sd_model) {
+        // Forge workaround
+        sd_model = gradioApp().getElementsByClassName("model_selection")[0]
+    }
    if (!sd_model.querySelector(".eta-bar")) {
        image_browser_webui_ready = true
        image_browser_start()