xinyun/ComfyUI-MochiWrapper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix encoding precision

Browse Source
This commit is contained in:
kijai 2024-11-01 18:21:34 +02:00
parent a5b06b02ad
commit ec298a1d64
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes.py
View File

@ -720,7 +720,7 @@ class MochiImageEncode:
images = images.unsqueeze(0) * 2 - 1 images = images.unsqueeze(0) * 2 - 1
images = rearrange(images, "t b h w c -> t c b h w") 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) print(images.shape)
encoder.to(device) encoder.to(device)
print("images before encoding", images.shape) print("images before encoding", images.shape)