From 4dcee21d126fa31159dedd01a655ff6c470251d7 Mon Sep 17 00:00:00 2001
From: matt3o <matt3o@gmail.com>
Date: Thu, 16 May 2024 13:09:25 +0200
Subject: [PATCH] faster mask blur

---
 essentials.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/essentials.py b/essentials.py
index c662a98..c950cdb 100644
--- a/essentials.py
+++ b/essentials.py
@@ -415,6 +415,7 @@ class MaskBlur:
     CATEGORY = "essentials"
 
     def execute(self, mask, amount):
+        mask = mask.to(comfy.model_management.get_torch_device())
         size = int(6 * amount + 1)
         if size % 2 == 0:
             size+= 1
@@ -425,6 +426,7 @@ class MaskBlur:
         blurred = mask.unsqueeze(1)
         blurred = T.GaussianBlur(size, amount)(blurred)
         blurred = blurred.squeeze(1)
+        blurred = blurred.to(comfy.model_management.intermediate_device())
 
         return(blurred,)