From 6183c00b95c8b46751acd438b20053dc3fafd497 Mon Sep 17 00:00:00 2001 From: "Dr.Lt.Data" Date: Mon, 15 Jan 2024 14:22:12 +0900 Subject: [PATCH] fix: invalid empty mask size related issue: https://github.com/ltdrdata/ComfyUI-Impact-Pack/issues/433 --- modules/impact/config.py | 2 +- modules/impact/core.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/impact/config.py b/modules/impact/config.py index 9004612..31d837d 100644 --- a/modules/impact/config.py +++ b/modules/impact/config.py @@ -2,7 +2,7 @@ import configparser import os -version_code = [4, 66, 2] +version_code = [4, 66, 3] version = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '') dependency_version = 20 diff --git a/modules/impact/core.py b/modules/impact/core.py index cc5e939..1c84293 100644 --- a/modules/impact/core.py +++ b/modules/impact/core.py @@ -607,7 +607,7 @@ def make_sam_mask(sam_model, segs, image, detection_hint, dilation, mask = dilate_mask(mask.cpu().numpy(), dilation) mask = torch.from_numpy(mask) else: - size = image.shape[0] // 8, image.shape[1] // 8 + size = image.shape[0], image.shape[1] mask = torch.zeros(size, dtype=torch.float32, device="cpu") # empty mask mask = utils.make_3d_mask(mask)