Removed self.embedding_identifiers from SDTokenizer.__init__, leaving self.embedding_identifier as the singular source of truth. Added dynamic embedding_identifiers property.

This commit is contained in:
silveroxides 2026-07-04 01:24:46 +02:00
parent 225660867a
commit 7d782329d3

View File

@ -535,12 +535,19 @@ class SDTokenizer:
self.embedding_directory = embedding_directory
self.max_word_length = 8
self.embedding_identifier = "embedding:"
self.embedding_identifiers = ["embedding:", "EMB:", "TE:"]
self.embedding_size = embedding_size
self.embedding_key = embedding_key
self.disable_weights = disable_weights
@property
def embedding_identifiers(self):
identifiers = [self.embedding_identifier]
for item in ["EMB:", "TE:"]:
if item not in identifiers:
identifiers.append(item)
return identifiers
def _try_get_embedding(self, embedding_name:str):
'''
Takes a potential embedding name and tries to retrieve it.