fix input image range

This commit is contained in:
kijai 2025-01-22 09:23:17 +02:00
parent 7242e7e22e
commit 2286d0c0f7

View File

@ -381,6 +381,7 @@ class Hy3DGenerateMesh:
offload_device = mm.unet_offload_device()
image = image.permute(0, 3, 1, 2).to(device)
image = image * 2 - 1
if mask is not None:
mask = mask.unsqueeze(0).to(device)