wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-19 21:08:19 +08:00
Code Issues Actions 24 Packages Projects Releases Wiki Activity

Merge c247f900aa into 1d1099bea0

Browse Source
This commit is contained in:
mannydelrio2 2026-07-17 12:59:40 -07:00 committed by GitHub
commit f75e7acac2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_train.py
View File

@ -923,7 +923,7 @@ def _run_training_loop(
if bucket_mode: if bucket_mode:
# Use first bucket's first latent as dummy for guider # Use first bucket's first latent as dummy for guider
dummy_latent = latents[0][:1].repeat(num_images, 1, 1, 1) dummy_latent = latents[0][:1].repeat(num_images, *([1] * (latents[0].dim() - 1)))
guider.sample( guider.sample(
noise.generate_noise({"samples": dummy_latent}), noise.generate_noise({"samples": dummy_latent}),
dummy_latent, dummy_latent,
@ -933,7 +933,7 @@ def _run_training_loop(
) )
elif multi_res: elif multi_res:
# use first latent as dummy latent if multi_res # use first latent as dummy latent if multi_res
latents = latents[0].repeat(num_images, 1, 1, 1) latents = latents[0].repeat(num_images, *([1] * (latents[0].dim() - 1)))
guider.sample( guider.sample(
noise.generate_noise({"samples": latents}), noise.generate_noise({"samples": latents}),
latents, latents,