xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-12 02:53:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed a bug and added support for more face images

Browse Source
This commit is contained in:
zhangsongbo 2024-09-09 20:25:42 +08:00
parent 0c91384121
commit 1ed629a567
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_photomaker_v2.py
View File

@ -236,7 +236,7 @@ class PhotoMakerEncodeV2:
if index > 0: if index > 0:
id_embeds = self.get_id_embeds(image) id_embeds = self.get_id_embeds(image)
token_index = index - 1 token_index = index - 1
num_id_images = 1 num_id_images = image.shape[0]
num_tokens_v2 = 2 num_tokens_v2 = 2
id_embeds = id_embeds.unsqueeze(0).to(device=photomaker.load_device) id_embeds = id_embeds.unsqueeze(0).to(device=photomaker.load_device)
class_tokens_mask = [True if token_index <= i < token_index+(num_id_images*num_tokens_v2) else False for i in range(cond.shape[1])] class_tokens_mask = [True if token_index <= i < token_index+(num_id_images*num_tokens_v2) else False for i in range(cond.shape[1])]
@ -248,7 +248,7 @@ class PhotoMakerEncodeV2:
return ([[out, {"pooled_output": pooled}]], [[cond, {"pooled_output": pooled}]],) return ([[out, {"pooled_output": pooled}]], [[cond, {"pooled_output": pooled}]],)
def tensor2np(tensor: torch.Tensor): def tensor2np(self, tensor: torch.Tensor):
if len(tensor.shape) == 3: # Single image if len(tensor.shape) == 3: # Single image
return np.clip(255.0 * tensor.cpu().numpy(), 0, 255).astype(np.uint8) return np.clip(255.0 * tensor.cpu().numpy(), 0, 255).astype(np.uint8)
else: # Batch of images else: # Batch of images