FIX: MaskHelper - move result back to CPU

To avoid "RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu" might be thrown by the next node.
2025-10-03 01:41:22 +07:00
parent a2ad854a09
commit 13d836015b
1 changed files with 1 additions and 0 deletions
@@ -1474,6 +1474,7 @@ class MaskHelper:
                face_segment[...,3] = mask[i]

                result = rgba2rgb_tensor(result)
+                result = result.cpu()  # Перемещаем результат обратно на CPU

                pbar.update(1)