From ec298a1d64aa8ca126ed788366ae421746d41c8d Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 1 Nov 2024 18:21:34 +0200
Subject: [PATCH] fix encoding precision

---
 nodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index c5d9eaa..69c2722 100644
--- a/nodes.py
+++ b/nodes.py
@@ -720,7 +720,7 @@ class MochiImageEncode:
         
         images = images.unsqueeze(0) * 2 - 1
         images = rearrange(images, "t b h w c -> t c b h w")
-        images = images.to(encoder.dtype).to(device)
+        images = images.to(device)
         print(images.shape)
         encoder.to(device)
         print("images before encoding", images.shape)