cxllm-vendors/stable-diffusion-webui-forge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

make Textual Inversion UI isolated

Browse Source 
so that we can replace it soon
also removed the extremely annoying SD version filter
This commit is contained in:
layerdiffusion
2024-08-03 21:21:41 -07:00
parent 2d30eeedfd
commit 7c612a012b
2 changed files with 16 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/textual_inversion/textual_inversion.py
+3 -1
View File
@@ -209,7 +209,7 @@ class EmbeddingDatabase:
errors.report(f"Error loading embedding {fn}", exc_info=True) errors.report(f"Error loading embedding {fn}", exc_info=True)
continue continue
def load_textual_inversion_embeddings(self, force_reload=False): def load_textual_inversion_embeddings(self, force_reload=False, sync_with_sd_model=True):
if not force_reload: if not force_reload:
need_reload = False need_reload = False
for embdir in self.embedding_dirs.values(): for embdir in self.embedding_dirs.values():
@@ -223,6 +223,8 @@ class EmbeddingDatabase:
self.ids_lookup.clear() self.ids_lookup.clear()
self.word_embeddings.clear() self.word_embeddings.clear()
self.skipped_embeddings.clear() self.skipped_embeddings.clear()
if sync_with_sd_model:
self.expected_shape = self.get_expected_shape() self.expected_shape = self.get_expected_shape()
for embdir in self.embedding_dirs.values(): for embdir in self.embedding_dirs.values():
modules/ui_extra_networks_textual_inversion.py
+12 -5
View File
@@ -1,19 +1,26 @@
import os import os
import modules.textual_inversion.textual_inversion
from modules import ui_extra_networks, sd_hijack, shared from modules.shared import cmd_opts
from modules import ui_extra_networks, shared
from modules.ui_extra_networks import quote_js from modules.ui_extra_networks import quote_js
embedding_db = modules.textual_inversion.textual_inversion.EmbeddingDatabase()
embedding_db.add_embedding_dir(cmd_opts.embeddings_dir)
embedding_db.load_textual_inversion_embeddings(force_reload=True, sync_with_sd_model=False)
class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage): class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage):
def __init__(self): def __init__(self):
super().__init__('Textual Inversion') super().__init__('Textual Inversion')
self.allow_negative_prompt = True self.allow_negative_prompt = True
def refresh(self): def refresh(self):
sd_hijack.model_hijack.embedding_db.load_textual_inversion_embeddings(force_reload=True) embedding_db.load_textual_inversion_embeddings(force_reload=True, sync_with_sd_model=False)
def create_item(self, name, index=None, enable_filter=True): def create_item(self, name, index=None, enable_filter=True):
embedding = sd_hijack.model_hijack.embedding_db.word_embeddings.get(name) embedding = embedding_db.word_embeddings.get(name)
if embedding is None: if embedding is None:
return return
@@ -35,11 +42,11 @@ class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage):
def list_items(self): def list_items(self):
# instantiate a list to protect against concurrent modification # instantiate a list to protect against concurrent modification
names = list(sd_hijack.model_hijack.embedding_db.word_embeddings) names = list(embedding_db.word_embeddings)
for index, name in enumerate(names): for index, name in enumerate(names):
item = self.create_item(name, index) item = self.create_item(name, index)
if item is not None: if item is not None:
yield item yield item
def allowed_directories_for_previews(self): def allowed_directories_for_previews(self):
return list(sd_hijack.model_hijack.embedding_db.embedding_dirs) return list(embedding_db.embedding_dirs)