Skip to content

Commit ecdc869

Browse files
Qwen Image Lora training fix from #11090 (#11094)
1 parent dce518c commit ecdc869

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy_extras/nodes_train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -623,7 +623,7 @@ def loss_callback(loss):
623623
noise = comfy_extras.nodes_custom_sampler.Noise_RandomNoise(seed)
624624
if multi_res:
625625
# use first latent as dummy latent if multi_res
626-
latents = latents[0].repeat(num_images, 1, 1, 1)
626+
latents = latents[0].repeat((num_images,) + ((1,) * (latents[0].ndim - 1)))
627627
guider.sample(
628628
noise.generate_noise({"samples": latents}),
629629
latents,

0 commit comments

Comments
 (0)