Added ability to set the caption extention in dataset viewer, captioner, and trainer so one dataset can have multiple caption styles in different files with different extensions. Added dataset caption template for a blank ideogram 4 formatted template.

This commit is contained in:
Jaret Burkett committed 2026-06-06 08:32:24 -06:00
1 parent 10cdeb394e
commit 41157b460c
19 files changed
+219 -270

No files matched your search

@@ -15,7 +15,6 @@ from toolkit.dataloader_mixins import (
LatentCachingFileItemDTOMixin,
ControlFileItemDTOMixin,
ArgBreakMixin,
PoiFileItemDTOMixin,
MaskFileItemDTOMixin,
AugmentationFileItemDTOMixin,
UnconditionalFileItemDTOMixin,
@@ -51,7 +50,6 @@ class FileItemDTO(
MaskFileItemDTOMixin,
AugmentationFileItemDTOMixin,
UnconditionalFileItemDTOMixin,
PoiFileItemDTOMixin,
ArgBreakMixin,
):
def __init__(self, *args, **kwargs):