Merge branch 'AlUlkesh:main' into main

2024-08-15 12:23:31 +08:00 · 2024-08-15 12:23:31 +08:00 · 6f1c2d2523
parent 3829185fb7 f353bc1152
commit 6f1c2d2523
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()