Fix BucketBatchSampler cache alignment in DreamBooth scripts

examples/dreambooth/train_dreambooth_lora_flux2.py

@@ -1005,8 +1005,7 @@ def __init__(self, dataset: DreamBoothDataset, batch_size: int, drop_last: bool
                 self.sampler_len += 1  # Count the number of batches
 
     def __iter__(self):
-        # Shuffle the order of the batches each epoch
-        random.shuffle(self.batches)
+        # Keep the precomputed batch order stable so step-indexed caches stay aligned.
         for batch in self.batches:
             yield batch
 

examples/dreambooth/train_dreambooth_lora_flux2_img2img.py

@@ -1003,8 +1003,7 @@ def __init__(self, dataset: DreamBoothDataset, batch_size: int, drop_last: bool
                 self.sampler_len += 1  # Count the number of batches
 
     def __iter__(self):
-        # Shuffle the order of the batches each epoch
-        random.shuffle(self.batches)
+        # Keep the precomputed batch order stable so step-indexed caches stay aligned.
         for batch in self.batches:
             yield batch
 

examples/dreambooth/train_dreambooth_lora_flux2_klein.py

@@ -1000,8 +1000,7 @@ def __init__(self, dataset: DreamBoothDataset, batch_size: int, drop_last: bool
                 self.sampler_len += 1  # Count the number of batches
 
     def __iter__(self):
-        # Shuffle the order of the batches each epoch
-        random.shuffle(self.batches)
+        # Keep the precomputed batch order stable so step-indexed caches stay aligned.
         for batch in self.batches:
             yield batch
 

examples/dreambooth/train_dreambooth_lora_flux2_klein_img2img.py

@@ -999,8 +999,7 @@ def __init__(self, dataset: DreamBoothDataset, batch_size: int, drop_last: bool
                 self.sampler_len += 1  # Count the number of batches
 
     def __iter__(self):
-        # Shuffle the order of the batches each epoch
-        random.shuffle(self.batches)
+        # Keep the precomputed batch order stable so step-indexed caches stay aligned.
         for batch in self.batches:
             yield batch
 

examples/dreambooth/train_dreambooth_lora_z_image.py

@@ -994,8 +994,7 @@ def __init__(self, dataset: DreamBoothDataset, batch_size: int, drop_last: bool
                 self.sampler_len += 1  # Count the number of batches
 
     def __iter__(self):
-        # Shuffle the order of the batches each epoch
-        random.shuffle(self.batches)
+        # Keep the precomputed batch order stable so step-indexed caches stay aligned.
         for batch in self.batches:
             yield batch