xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-14 01:46:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update latent_preview.py

Browse Source
This commit is contained in:
HGB 2024-11-18 00:21:14 +07:00 committed by GitHub
parent 420000d608
commit 39c318ee21
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
latent_preview.py
View File

@ -144,9 +144,12 @@ def prepare_callback(model, steps, x0_output_dict=None):
if previewer:
preview_bytes = previewer.decode_latent_to_preview_image(preview_format, x0)
pbar.update_absolute(step + 1, total_steps, preview_bytes)
if args.xla or args.xla_fsdp:
if args.xla or args.xla_spmd:
import torch_xla as xla
xla.sync()
pbar.update_absolute(step + 1, total_steps, preview_bytes)
return callback