nodes/Jokimbe-ComfyUI-DrawThings-gRPC
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
Jo Kim
2025-04-07 19:06:44 +02:00
parent 6364dd6431
commit b9f3eb2c1d
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+8 -5
View File
@@ -41,14 +41,17 @@ def prepare_callback(step, total_steps, x0: torch.Tensor, latent_format):
if preview_format not in ["JPEG", "PNG"]:
preview_format = "JPEG"
if x0 is not None:
previewer = latent_preview.get_previewer(x0.device, latent_format)
pbar = comfy.utils.ProgressBar(step)
def callback(step, x0: torch.Tensor, total_steps, latent_format):
def callback(step, x0: torch.Tensor, total_steps):
preview_bytes = None
if x0 is not None:
previewer = latent_preview.get_previewer(x0.device, latent_format)
if previewer:
preview_bytes = previewer.decode_latent_to_preview_image(preview_format, x0)
pbar.update_absolute(step, total_steps, preview_bytes)
return callback(step, x0, total_steps, latent_format)
pbar.update_absolute(step + 1, total_steps, preview_bytes)
return callback(step, x0, total_steps)
# def image_to_base64(image_tensor: torch.Tensor):
# if image_tensor is not None: