From 4b43c155c319a8fb79ea4b62460e4c92dea617e8 Mon Sep 17 00:00:00 2001
From: risunobushi <113669885+risunobushi@users.noreply.github.com>
Date: Wed, 29 Jan 2025 15:14:03 +0100
Subject: [PATCH] Update logo_displacement_node.py

---
 logo_displacement_node.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/logo_displacement_node.py b/logo_displacement_node.py
index 7c107ba..18d2c28 100644
--- a/logo_displacement_node.py
+++ b/logo_displacement_node.py
@@ -59,10 +59,10 @@ class DisplaceLogo:
         new_y = new_y / (height - 1) * 2 - 1
 
         # Stack to create grid tensor
-        grid = torch.stack([torch.tensor(new_x), torch.tensor(new_y)], dim=-1).unsqueeze(0)
+        grid = torch.stack([torch.tensor(new_x, dtype=torch.float32), torch.tensor(new_y, dtype=torch.float32)], dim=-1).unsqueeze(0)
 
         # Convert logo to tensor and apply grid_sample
-        logo_tensor = torch.from_numpy(logo).permute(2, 0, 1).unsqueeze(0)
+        logo_tensor = torch.from_numpy(logo).permute(2, 0, 1).unsqueeze(0).float()
         displaced_logo = F.grid_sample(logo_tensor, grid, mode='bilinear', padding_mode='border', align_corners=True)
 
         # Convert back to ComfyUI format