Fix model change detection in newest gradio

Fixes #172
This commit is contained in:
DominikDoom
2023-05-05 11:28:01 +02:00
parent 4442cb78ec
commit ba38d1b893

View File

@@ -870,13 +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 span.single-select")
let modelDropdown = gradioApp().querySelector("#setting_sd_model_checkpoint input") || gradioApp().querySelector("#setting_sd_model_checkpoint select");
if (modelDropdown) {
currentModelName = modelDropdown.textContent;
} else {
// Fallback for older versions
modelDropdown = gradioApp().querySelector("#setting_sd_model_checkpoint select");
currentModelName = modelDropdown.value;
} else {
// Fallback for intermediate versions
modelDropdown = gradioApp().querySelector("#setting_sd_model_checkpoint span.single-select");
currentModelName = modelDropdown.textContent;
}
}
}