From e374bd304a4a5b44d0deae2b2afaf4d5faad4810 Mon Sep 17 00:00:00 2001
From: "Dr.Lt.Data" <dr.lt.data@gmail.com>
Date: Mon, 11 Mar 2024 22:18:42 +0900
Subject: [PATCH] fix: different device error on SEGSPaste

---
 modules/impact/config.py     | 2 +-
 modules/impact/segs_nodes.py | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/modules/impact/config.py b/modules/impact/config.py
index a067f4b..4dc411e 100644
--- a/modules/impact/config.py
+++ b/modules/impact/config.py
@@ -2,7 +2,7 @@ import configparser
 import os
 
 
-version_code = [4, 83, 2]
+version_code = [4, 83, 3]
 version = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')
 
 dependency_version = 20
diff --git a/modules/impact/segs_nodes.py b/modules/impact/segs_nodes.py
index 220d000..ce24706 100644
--- a/modules/impact/segs_nodes.py
+++ b/modules/impact/segs_nodes.py
@@ -177,6 +177,12 @@ class SEGSPaste:
 
                     mask = tensor_gaussian_blur_mask(mask, feather) * (alpha/255)
                     x, y, *_ = seg.crop_region
+
+                    # ensure same device
+                    mask.cpu()
+                    image_i.cpu()
+                    ref_image.cpu()
+
                     tensor_paste(image_i, ref_image, (x, y), mask)
 
             if result is None: