From 42f126c79e973c34b808fd7c65f1be6b37280ed5 Mon Sep 17 00:00:00 2001 From: nkchocoai Date: Sat, 15 Jun 2024 12:30:36 +0900 Subject: [PATCH] Fix SD3 error --- py/defs/formatters.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/py/defs/formatters.py b/py/defs/formatters.py index 0c86031..074e2b8 100644 --- a/py/defs/formatters.py +++ b/py/defs/formatters.py @@ -78,7 +78,8 @@ def _extract_embedding_names(text, input_data): for word in to_tokenize: # if we find an embedding, deal with the embedding if ( - word.startswith(tokenizer.embedding_identifier) + hasattr(tokenizer, "embedding_identifier") + and word.startswith(tokenizer.embedding_identifier) and tokenizer.embedding_directory is not None ): embedding_name = word[len(tokenizer.embedding_identifier) :].strip("\n")