mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-07-08 15:37:46 +08:00
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:
parent
225660867a
commit
7d782329d3
@ -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.
|
||||
|
||||
Loading…
Reference in New Issue
Block a user