fix encoding precision

2024-11-01 18:21:34 +02:00 · 2024-11-01 18:21:34 +02:00 · ec298a1d64
commit ec298a1d64
parent a5b06b02ad
1 changed files with 1 additions and 1 deletions
--- 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)