From 65bc2c0327be3e1b200237357becbdc33d5a6cf4 Mon Sep 17 00:00:00 2001
From: Michael Poutre <m1kep.my.mail@gmail.com>
Date: Wed, 13 Sep 2023 19:00:33 -0700
Subject: [PATCH] refactor(Node): Use new
 comfy.supported_models_base.ClipTarget

---
 nodes.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/nodes.py b/nodes.py
index 37d8706..6a3e6ee 100644
--- a/nodes.py
+++ b/nodes.py
@@ -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(