fix input image range

2025-12-10 05:14:29 +08:00 · 2025-01-22 09:23:17 +02:00 · 2025-01-22 09:23:17 +02:00 · 2286d0c0f7
commit 2286d0c0f7
parent 7242e7e22e
1 changed files with 1 additions and 0 deletions
--- a/nodes.py
+++ b/nodes.py
@ -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)