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,)