Merge pull request #1862 from bghira/bugfix/conditioning-image-embed-dataset

fix bug when loading predefined conditioning_image_embeds sets

Author: codecadede

simpletuner/helpers/data_backend/config/base.py

@@ -53,6 +53,7 @@ def _validate_common_fields(self, args: Dict[str, Any]) -> None:
             DatasetType.VIDEO,
             DatasetType.TEXT_EMBEDS,
             DatasetType.IMAGE_EMBEDS,
+            DatasetType.CONDITIONING_IMAGE_EMBEDS,
             DatasetType.CAPTION,
         ]
         if self.dataset_type not in valid_dataset_types:

simpletuner/helpers/data_backend/config/image_embed.py

@@ -45,7 +45,11 @@ def apply_defaults(self, args: Dict[str, Any]) -> None:
     def validate(self, args: Dict[str, Any]) -> None:
         validators.validate_backend_id(self.id)
 
-        validators.validate_dataset_type(self.dataset_type, [DatasetType.IMAGE_EMBEDS], self.id)
+        validators.validate_dataset_type(
+            self.dataset_type,
+            [DatasetType.IMAGE_EMBEDS, DatasetType.CONDITIONING_IMAGE_EMBEDS],
+            self.id,
+        )
 
         validators.check_for_caption_filter_list_misuse(self.dataset_type, False, self.id)