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-11-04 15:10:15 +02:00
parent f94cf43331
commit 78f9e7b896
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

@ -634,7 +634,7 @@ class MochiDecode:
return torch.cat(result_rows, dim=3)
vae.to(device)
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 and frame_batch_size > T:
logging.warning(f"Frame batch size is larger than the number of samples, setting to {T}")
frame_batch_size = T