jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues 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

5
comfy/sd.py
View File

@@ -252,18 +252,15 @@ class CLIP:
else:
params = {}
tokenizer_params = {}
if self.target_clip == "ldm.modules.encoders.modules.FrozenOpenCLIPEmbedder":
clip = sd2_clip.SD2ClipModel
tokenizer = sd2_clip.SD2Tokenizer
elif self.target_clip == "ldm.modules.encoders.modules.FrozenCLIPEmbedder":
clip = sd1_clip.SD1ClipModel
tokenizer = sd1_clip.SD1Tokenizer
tokenizer_params['embedding_directory'] = embedding_directory
self.cond_stage_model = clip(**(params))
self.tokenizer = tokenizer(**(tokenizer_params))
self.tokenizer = tokenizer(embedding_directory=embedding_directory)
self.patcher = ModelPatcher(self.cond_stage_model)
def clone(self):