Skip to content

Commit 73175cf

Browse files
committed
split tokenizer from encoder
1 parent 8489cba commit 73175cf

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

comfy/sd.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -372,10 +372,12 @@ def add_patches(self, patches, strength=1.0):
372372
def clip_layer(self, layer_idx):
373373
self.layer_idx = layer_idx
374374

375-
def encode(self, text):
375+
def tokenize(self, text):
376+
return self.tokenizer.tokenize_with_weights(text)
377+
378+
def encode(self, tokens):
376379
if self.layer_idx is not None:
377380
self.cond_stage_model.clip_layer(self.layer_idx)
378-
tokens = self.tokenizer.tokenize_with_weights(text)
379381
try:
380382
self.patcher.patch_model()
381383
cond = self.cond_stage_model.encode_token_weights(tokens)

nodes.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,8 @@ def INPUT_TYPES(s):
4444
CATEGORY = "conditioning"
4545

4646
def encode(self, clip, text):
47-
return ([[clip.encode(text), {}]], )
47+
tokens = clip.tokenize(text)
48+
return ([[clip.encode(tokens), {}]], )
4849

4950
class ConditioningCombine:
5051
@classmethod

0 commit comments

Comments
 (0)