mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
Fix naming, use Exception as parent class
This commit is contained in:
@@ -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,4 +1,4 @@
|
||||
class ErrorLimiterTrigger(RuntimeError):
|
||||
class ErrorLimiterTrigger(Exception):
|
||||
def __init__(self, name: str, *args):
|
||||
super().__init__(*args)
|
||||
self.name = name
|
||||
|
||||
Reference in New Issue
Block a user