diff --git a/modules/textual_inversion.py b/modules/textual_inversion.py index 0605340e1..9433130f0 100644 --- a/modules/textual_inversion.py +++ b/modules/textual_inversion.py @@ -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): diff --git a/sdnext_core/errorlimiter.py b/sdnext_core/errorlimiter.py index 4f2797d1c..ea8afd1e5 100644 --- a/sdnext_core/errorlimiter.py +++ b/sdnext_core/errorlimiter.py @@ -1,4 +1,4 @@ -class ErrorLimiterTrigger(RuntimeError): +class ErrorLimiterTrigger(Exception): def __init__(self, name: str, *args): super().__init__(*args) self.name = name