fix bug caused by refresh not being called properly

This commit is contained in:
Sj-Si 2024-04-15 12:37:04 -04:00
parent 8ca069e7dd
commit 634f7bc920
6 changed files with 13 additions and 0 deletions

View File

@ -14,6 +14,7 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
def refresh(self):
networks.list_available_networks()
super().refresh()
def create_item(self, name, index=None, enable_filter=True):
lora_on_disk = networks.available_networks.get(name)

View File

@ -337,6 +337,8 @@ class ExtraNetworksTab {
}
async onFetchCardsData(div_ids) {
await this.waitForServerPageReady();
return JSON.parse(
await requestGetPromise(
"./sd_extra_networks/fetch-cards-data",
@ -349,6 +351,8 @@ class ExtraNetworksTab {
}
async onFetchTreeData(div_ids) {
await this.waitForServerPageReady();
return JSON.parse(
await requestGetPromise(
"./sd_extra_networks/fetch-tree-data",

View File

@ -201,7 +201,12 @@ def fetch_cover_images(extra_networks_tabname: str = "", item: str = "", index:
def init_tree_data(tabname: str = "", extra_networks_tabname: str = "") -> JSONResponse:
page = get_page_by_name(extra_networks_tabname)
data = page.generate_tree_view_data(tabname)
if data is None:
return JSONResponse({}, status_code=503)
return JSONResponse(data, status_code=200)
def fetch_tree_data(

View File

@ -13,6 +13,7 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
def refresh(self):
shared.refresh_checkpoints()
super().refresh()
def create_item(self, name, index=None, enable_filter=True):
checkpoint: sd_models.CheckpointInfo = sd_models.checkpoint_aliases.get(name)

View File

@ -11,6 +11,7 @@ class ExtraNetworksPageHypernetworks(ui_extra_networks.ExtraNetworksPage):
def refresh(self):
shared.reload_hypernetworks()
super().refresh()
def create_item(self, name, index=None, enable_filter=True):
full_path = shared.hypernetworks.get(name)

View File

@ -11,6 +11,7 @@ class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage):
def refresh(self):
sd_hijack.model_hijack.embedding_db.load_textual_inversion_embeddings(force_reload=True)
super().refresh()
def create_item(self, name, index=None, enable_filter=True):
embedding = sd_hijack.model_hijack.embedding_db.word_embeddings.get(name)