fix: crash when using a non-square mask

This commit is contained in:
Kelly Jerrell
2025-05-25 10:08:08 -07:00
parent 8e65df2bfb
commit 01fb7f0fa9
2 changed files with 4 additions and 3 deletions
+1 -1
View File
@@ -288,7 +288,7 @@ def convert_mask_for_request(mask_tensor: torch.Tensor, image_tensor: torch.Tens
CCV_TENSOR_FORMAT_NCHW,
CCV_8U,
0,
height, width, 0, 0
width, height, 0, 0
)
for y in range(height):
+3 -2
View File
@@ -4,10 +4,11 @@ build-backend = "setuptools.build_meta"
[project]
name = "DrawThings-gRPC"
version = "0.0.4"
version = "0.0.5"
description = "Connect to a Draw Things gRPC server"
authors = [
{name = "Jokimbe"}
{name = "Jokimbe"},
{name = "kcjerrell"},
]
readme = "README.md"
license = {text = "GNU General Public License v3"}