xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2026-05-02 22:20:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Error on too long prompts.

Browse Source
This commit is contained in:
kijai 2024-10-13 15:38:06 +03:00
parent a995247624
commit 42c46b76e1
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
nodes.py
View File

@ -751,14 +751,18 @@ class CogVideoTextEncode:
CATEGORY = "CogVideoWrapper" CATEGORY = "CogVideoWrapper"
def process(self, clip, prompt, strength=1.0, force_offload=True): def process(self, clip, prompt, strength=1.0, force_offload=True):
max_tokens = 226
load_device = mm.text_encoder_device() load_device = mm.text_encoder_device()
offload_device = mm.text_encoder_offload_device() offload_device = mm.text_encoder_offload_device()
clip.tokenizer.t5xxl.pad_to_max_length = True clip.tokenizer.t5xxl.pad_to_max_length = True
clip.tokenizer.t5xxl.max_length = 226 clip.tokenizer.t5xxl.max_length = max_tokens
clip.cond_stage_model.to(load_device) clip.cond_stage_model.to(load_device)
tokens = clip.tokenize(prompt, return_word_ids=True) tokens = clip.tokenize(prompt, return_word_ids=True)
embeds = clip.encode_from_tokens(tokens, return_pooled=False, return_dict=False) embeds = clip.encode_from_tokens(tokens, return_pooled=False, return_dict=False)
if embeds.shape[1] > 226:
raise ValueError(f"Prompt is too long, max tokens supported is {max_tokens} or less, got {embeds.shape[1]}")
embeds *= strength embeds *= strength
if force_offload: if force_offload:
clip.cond_stage_model.to(offload_device) clip.cond_stage_model.to(offload_device)