From 896f0a3531757fd8bebb70a85b0f29d868b6a1d6 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Tue, 22 Apr 2025 08:54:26 +0300 Subject: [PATCH] Update nodes.py --- nodes.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/nodes.py b/nodes.py index 95446ac..fb49585 100644 --- a/nodes.py +++ b/nodes.py @@ -1229,14 +1229,22 @@ class Hy3DGenerateMeshMultiView(): pipeline.to(device) - if front is not None: + if front is not None and not torch.all(front < 1e-6).item(): front = front.clone().permute(0, 3, 1, 2).to(device) - if back is not None: + else: + front = None + if back is not None and not torch.all(back < 1e-6).item(): back = back.clone().permute(0, 3, 1, 2).to(device) - if left is not None: + else: + back = None + if left is not None and not torch.all(left < 1e-6).item(): left = left.clone().permute(0, 3, 1, 2).to(device) - if right is not None: + else: + left = None + if right is not None and not torch.all(right < 1e-6).item(): right = right.clone().permute(0, 3, 1, 2).to(device) + else: + right = None view_dict = { 'front': front,