From 89e4ac52d0bab4ae4d5e3e40786ca54bdb0ebfd8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E9=9B=AA=E5=B3=B0?= Date: Wed, 22 Jan 2025 16:26:47 +0800 Subject: [PATCH] fix mask shape --- PulidFluxHook.py | 4 ++-- pyproject.toml | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/PulidFluxHook.py b/PulidFluxHook.py index a01ed90..f750920 100644 --- a/PulidFluxHook.py +++ b/PulidFluxHook.py @@ -40,8 +40,8 @@ def pulid_patch(img, pulid_model=None, ca_idx=None, weight=1.0, embedding=None, patch_size = transformer_options[PatchKeys.running_net_model].patch_size mask = comfy.ldm.common_dit.pad_to_patch_size(mask, (patch_size, patch_size)) mask = rearrange(mask, "b c (h ph) (w pw) -> b (h w) (c ph pw)", ph=patch_size, pw=patch_size) - # (b, seq_len, _) =>(b, seq_len, seq_len) - mask = mask[..., 0].unsqueeze(-1).repeat(1, 1, mask.shape[1]).to(dtype=pulid_img.dtype) + # (b, seq_len, _) =>(b, seq_len, pulid.dim) + mask = mask[..., 0].unsqueeze(-1).repeat(1, 1, pulid_img.shape[-1]).to(dtype=pulid_img.dtype) del patch_size, latent_image_shape pulid_img = pulid_img * mask diff --git a/pyproject.toml b/pyproject.toml index 82a0453..31405bb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui_pulid_flux_ll" description = "The implementation for PuLID-Flux, support use with TeaCache and WaveSpeed, no model pollution." -version = "1.0.8" +version = "1.0.9" license = {file = "LICENSE"} dependencies = ['facexlib', 'insightface', 'onnxruntime', 'onnxruntime-gpu; sys_platform != "darwin" and platform_machine == "x86_64"', 'ftfy', 'timm']