Add custom seperator

This commit is contained in:
Kohaku-Blueleaf
2023-10-30 19:55:30 +08:00
parent 202f2c3292
commit 5dc2a0d3fd
2 changed files with 14 additions and 2 deletions

View File

@@ -51,6 +51,7 @@ class BaseSubsetParams:
image_dir: Optional[str] = None
num_repeats: int = 1
shuffle_caption: bool = False
caption_seperator: str = ',',
keep_tokens: int = 0
color_aug: bool = False
flip_aug: bool = False