Fix naming, use Exception as parent class

This commit is contained in:
awsr
2026-01-22 17:25:42 -08:00
parent 6700edaf7d
commit 0310dc8fd6
2 changed files with 5 additions and 5 deletions
+4 -4
View File
@@ -272,7 +272,7 @@ class EmbeddingDatabase:
text_encoders, tokenizers, hiddensizes = get_text_encoders()
if not all([text_encoders, tokenizers, hiddensizes]):
return
ErrorLimiter.start(self.__qualname__)
ErrorLimiter.start("load_diffusers_embedding_1")
for embedding in embeddings:
try:
embedding.vector_sizes = [v.shape[-1] for v in embedding.vec]
@@ -287,14 +287,14 @@ class EmbeddingDatabase:
except Exception as e:
shared.log.error(f'Load embedding invalid: name="{embedding.name}" fn="{filename}" {e}')
self.skipped_embeddings[embedding.name] = embedding
ErrorLimiter.update(self.__qualname__)
ErrorLimiter.update("load_diffusers_embedding_1")
if overwrite:
shared.log.info(f"Load bundled embeddings: {list(data.keys())}")
for embedding in embeddings:
if embedding.name not in self.skipped_embeddings:
deref_tokenizers(embedding.tokens, tokenizers)
insert_tokens(embeddings, tokenizers)
ErrorLimiter.start(self.__qualname__)
ErrorLimiter.start("load_diffusers_embedding_2")
for embedding in embeddings:
if embedding.name not in self.skipped_embeddings:
try:
@@ -303,7 +303,7 @@ class EmbeddingDatabase:
except Exception as e:
shared.log.error(f'Load embedding: name="{embedding.name}" file="{embedding.filename}" {e}')
errors.display(e, f'Load embedding: name="{embedding.name}" file="{embedding.filename}"')
ErrorLimiter.update(self.__qualname__)
ErrorLimiter.update("load_diffusers_embedding_2")
return
def load_from_dir(self, embdir):
+1 -1
View File
@@ -1,4 +1,4 @@
class ErrorLimiterTrigger(RuntimeError):
class ErrorLimiterTrigger(Exception):
def __init__(self, name: str, *args):
super().__init__(*args)
self.name = name