feat(ml):verify multi batch prompt match x

models/modules/img2img_turbo/img2img_turbo.py

@@ -192,7 +192,6 @@ def __init__(self, in_channels, out_channels, lora_rank_unet, lora_rank_vae):
         unet.enable_gradient_checkpointing()
 
     def forward(self, x, prompt):
-
         caption_tokens = self.tokenizer(
             prompt,
             max_length=self.tokenizer.model_max_length,
@@ -205,7 +204,7 @@ def forward(self, x, prompt):
         batch_size = caption_enc.shape[0]
         repeated_encs = [
             caption_enc[i].repeat(int(x.shape[0] / batch_size), 1, 1)
-            for i in range(caption_enc.shape[0])
+            for i in range(batch_size)
         ]
 
         # Concatenate the repeated encodings along the batch dimension