shenglin.net/ComfyUI-MochiWrapper-2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-10-30 22:08:22 +02:00
parent d971a19410
commit ebd0f62d53
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nodes.py
View File

@ -578,10 +578,9 @@ class MochiDecodeSpatialTiling:
B, C, T, H, W = samples.shape B, C, T, H, W = samples.shape
vae.to(device) vae.to(device)
decoded_list = [] decoded_list = []
with torch.autocast(mm.get_autocast_device(device), dtype=torch.bfloat16): with torch.autocast(mm.get_autocast_device(device), dtype=vae.dtype):
if enable_vae_tiling: if enable_vae_tiling:
from .mochi_preview.vae.model import apply_tiled from .mochi_preview.vae.model import apply_tiled