add cross-attention support to im2img; prevent inpainting from crashing

2024-08-30 20:32:17 +00:00 · 2022-10-18 23:23:38 +02:00
parent 2b79a716aa
commit 582880b314
4 changed files with 15 additions and 5 deletions
--- a/ldm/generate.py
+++ b/ldm/generate.py
@ -541,7 +541,8 @@ class Generate:
        image = Image.open(image_path)

        # used by multiple postfixers
-        uc, c = get_uc_and_c(
+        # todo: cross-attention
+        uc, c, _, _ = get_uc_and_c_and_ec(
            prompt, model =self.model,
            skip_normalize=opt.skip_normalize,
            log_tokens    =opt.log_tokenization