Make model dropdown selectors a bit more compact

This commit is contained in:
DominikDoom
2023-05-05 15:58:47 +02:00
parent ba38d1b893
commit 64cf9b2159

View File

@@ -870,12 +870,13 @@ async function setup() {
for (const mutation of mutationList) {
if (mutation.type === "attributes" && mutation.attributeName === "title") {
currentModelHash = mutation.target.title;
let modelDropdown = gradioApp().querySelector("#setting_sd_model_checkpoint input") || gradioApp().querySelector("#setting_sd_model_checkpoint select");
let sdm = gradioApp().querySelector("#setting_sd_model_checkpoint");
let modelDropdown = sdm.querySelector("input") || sdm.querySelector("#select");
if (modelDropdown) {
currentModelName = modelDropdown.value;
} else {
// Fallback for intermediate versions
modelDropdown = gradioApp().querySelector("#setting_sd_model_checkpoint span.single-select");
modelDropdown = sdm.querySelector("span.single-select");
currentModelName = modelDropdown.textContent;
}
}