We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dce518c commit ecdc869Copy full SHA for ecdc869
comfy_extras/nodes_train.py
@@ -623,7 +623,7 @@ def loss_callback(loss):
623
noise = comfy_extras.nodes_custom_sampler.Noise_RandomNoise(seed)
624
if multi_res:
625
# use first latent as dummy latent if multi_res
626
- latents = latents[0].repeat(num_images, 1, 1, 1)
+ latents = latents[0].repeat((num_images,) + ((1,) * (latents[0].ndim - 1)))
627
guider.sample(
628
noise.generate_noise({"samples": latents}),
629
latents,
0 commit comments