From 887de74d22b39d318de7865c617ed29047774cdc Mon Sep 17 00:00:00 2001 From: Disty0 Date: Fri, 31 May 2024 18:40:39 +0300 Subject: [PATCH] Fix Cascade overcooking with more than 75 tokens --- modules/prompt_parser_diffusers.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/prompt_parser_diffusers.py b/modules/prompt_parser_diffusers.py index 4d242458f..bfe267e2e 100644 --- a/modules/prompt_parser_diffusers.py +++ b/modules/prompt_parser_diffusers.py @@ -230,6 +230,7 @@ def pad_to_same_length(pipe, embeds): try: if getattr(pipe, "prior_pipe", None) and getattr(pipe.prior_pipe, "text_encoder", None) is not None: # Cascade empty_embed = pipe.prior_pipe.encode_prompt(device, 1, 1, False, "") + empty_embed = [torch.zeros(empty_embed[0].shape, device=empty_embed[0].device, dtype=empty_embed[0].dtype)] else: # SDXL empty_embed = pipe.encode_prompt("") except TypeError: # SD1.5