class_token first instead

This commit is contained in:
kijai
2024-08-24 18:48:04 +03:00
parent ec58e85b11
commit 9f52607be1
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -1775,7 +1775,7 @@ class DreamBoothDataset(BaseDataset):
else:
final_caption = cap_for_img
if subset.class_tokens:
final_caption += f" {subset.class_tokens}" # Concatenate class token to the caption
final_caption = f"{subset.class_tokens} {final_caption}" # Prepend class token to the caption
captions.append(final_caption)
+1 -1
View File
@@ -105,7 +105,7 @@ class TrainDatasetAdd:
"height": ("INT",{"min": 64, "default": 1024, "tooltip": "image height when bucketing is not used, also the default validation sampling height"}),
"batch_size": ("INT",{"min": 1, "default": 2, "tooltip": "Higher batch size uses more memory and generalizes the training more. "}),
"dataset_path": ("STRING",{"multiline": True, "default": "", "tooltip": "path to dataset, root is ComfyUI folder"}),
"class_tokens": ("STRING",{"multiline": True, "default": "", "tooltip": "aka trigger word, if specified, will be added to the end of each caption, if no captions exist, will be used on it's own"}),
"class_tokens": ("STRING",{"multiline": True, "default": "", "tooltip": "aka trigger word, if specified, will be added to the start of each caption, if no captions exist, will be used on it's own"}),
"enable_bucket": ("BOOLEAN",{"default": True, "tooltip": "enable buckets for multi aspect ratio training"}),
"bucket_no_upscale": ("BOOLEAN",{"default": False, "tooltip": "bucket reso is defined by image size automatically"}),
"num_repeats": ("INT", {"default": 1, "min": 1, "tooltip": "number of times to repeat dataset for an epoch"}),