refactor(Node): Use new comfy.supported_models_base.ClipTarget

This commit is contained in:
Michael Poutre
2023-11-13 19:47:56 -08:00
parent d67d8600ab
commit 65bc2c0327
+2 -4
View File
@@ -8,6 +8,7 @@ from PIL import Image
import folder_paths
import comfy.sd
import comfy.ops
from comfy.supported_models_base import ClipTarget
from custom_nodes.KepPromptLang.lib.clip_model import PromptLangClipModel
from custom_nodes.KepPromptLang.lib.tokenizer import PromptLangTokenizer
@@ -33,10 +34,7 @@ class SpecialClipLoader:
@staticmethod
def load_clip(source_clip: comfy.sd.CLIP) -> Tuple[comfy.sd.CLIP]:
clip_target = EmptyClass()
clip_target.params = {}
clip_target.clip = PromptLangClipModel
clip_target.tokenizer = PromptLangTokenizer
clip_target = ClipTarget(PromptLangTokenizer, PromptLangClipModel)
clip = comfy.sd.CLIP(clip_target, embedding_directory=source_clip.tokenizer.embedding_directory)
comfy.sd.load_clip_weights(