From f353bc1152ada7e59da6b88b9eef905ff67a58d0 Mon Sep 17 00:00:00 2001 From: AlUlkesh <99896447+AlUlkesh@users.noreply.github.com> Date: Wed, 14 Aug 2024 16:43:22 +0200 Subject: [PATCH] Fix for Forge, #258 --- javascript/image_browser.js | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/javascript/image_browser.js b/javascript/image_browser.js index f5f6da3..1c619fd 100644 --- 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()