diff --git a/javascript/civitai_helper.js b/javascript/civitai_helper.js index 7d7ebdf..0a5d0b0 100644 --- a/javascript/civitai_helper.js +++ b/javascript/civitai_helper.js @@ -182,6 +182,7 @@ onUiLoaded(() => { let model_name = ""; let model_type = ""; let cards = null; + let need_to_add_buttons = false; for (const tab_prefix of tab_prefix_list) { for (const js_model_type of model_type_list) { //get model_type for python side @@ -221,12 +222,17 @@ onUiLoaded(() => { replace_preview_btn = card.querySelector(".actions .additional a"); if (replace_preview_btn) { if (replace_preview_btn.innerHTML == "replace preview") { + need_to_add_buttons = true; replace_preview_btn.innerHTML = "🖼"; replace_preview_btn.style.margin = "0px 5px"; replace_preview_btn.style.fontSize = "200%"; } } + if (!need_to_add_buttons) { + continue; + } + //get model name node model_name_node = card.querySelector(".actions .name"); if (!model_name_node){