diff --git a/library/config_util.py b/library/config_util.py
index 3035bedb..d50d71f1 100644
--- a/library/config_util.py
+++ b/library/config_util.py
@@ -419,7 +419,8 @@ def generate_dataset_group_by_blueprint(dataset_group_blueprint: DatasetGroupBlu
   print(info)
 
   # make buckets first because it determines the length of dataset
-  for dataset in datasets:
+  for i, dataset in enumerate(datasets):
+    print(f"[Dataset {i}]")
     dataset.make_buckets()
 
   return DatasetGroup(datasets)
diff --git a/library/train_util.py b/library/train_util.py
index c377a56a..e5b81f1f 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1054,9 +1054,9 @@ class DatasetGroup(torch.utils.data.ConcatDataset):
     for dataset in self.datasets:
       dataset.add_replacement(str_from, str_to)
 
-  def make_buckets(self):
-    for dataset in self.datasets:
-      dataset.make_buckets()
+  # def make_buckets(self):
+  #   for dataset in self.datasets:
+  #     dataset.make_buckets()
 
   def cache_latents(self, vae):
     for dataset in self.datasets: