class_token first instead
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
@@ -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"}),
|
||||
|
||||
Reference in New Issue
Block a user