mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-13 21:16:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Embeddings/textual inversion support for SD2.x

Browse Source
This commit is contained in:
comfyanonymous
2023-02-05 15:49:03 -05:00
parent 1de5aa6a59
commit b1a7c9ebf6
2 changed files with 3 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/sd2_clip.py
View File

@@ -30,7 +30,5 @@ class SD2ClipModel(sd1_clip.SD1ClipModel):
self.layer_idx = layer_idx
class SD2Tokenizer(sd1_clip.SD1Tokenizer):
def __init__(self, tokenizer_path=None):
super().__init__(tokenizer_path, pad_with_end=False)
def __init__(self, tokenizer_path=None, embedding_directory=None):
super().__init__(tokenizer_path, pad_with_end=False, embedding_directory=embedding_directory)