wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-07 10:17:31 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Force min length 1 when tokenizing for text generation. (#12538)

Browse Source
This commit is contained in:
comfyanonymous 2026-02-19 19:57:44 -08:00 committed by GitHub
parent 0301ccf745
commit 5f2117528a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/sd1_clip.py
View File

@ -573,6 +573,8 @@ class SDTokenizer:
min_length = tokenizer_options.get("{}_min_length".format(self.embedding_key), self.min_length) min_length = tokenizer_options.get("{}_min_length".format(self.embedding_key), self.min_length)
min_padding = tokenizer_options.get("{}_min_padding".format(self.embedding_key), self.min_padding) min_padding = tokenizer_options.get("{}_min_padding".format(self.embedding_key), self.min_padding)
min_length = kwargs.get("min_length", min_length)
text = escape_important(text) text = escape_important(text)
if kwargs.get("disable_weights", self.disable_weights): if kwargs.get("disable_weights", self.disable_weights):
parsed_weights = [(text, 1.0)] parsed_weights = [(text, 1.0)]

2
comfy_extras/nodes_textgen.py
View File

@ -42,7 +42,7 @@ class TextGenerate(io.ComfyNode):
@classmethod @classmethod
def execute(cls, clip, prompt, max_length, sampling_mode, image=None) -> io.NodeOutput: def execute(cls, clip, prompt, max_length, sampling_mode, image=None) -> io.NodeOutput:
tokens = clip.tokenize(prompt, image=image, skip_template=False) tokens = clip.tokenize(prompt, image=image, skip_template=False, min_length=1)
# Get sampling parameters from dynamic combo # Get sampling parameters from dynamic combo
do_sample = sampling_mode.get("sampling_mode") == "on" do_sample = sampling_mode.get("sampling_mode") == "on"