v2 / Regainer-v2_config /dataset_config.toml
ProjectNekoFi's picture
feat: upload Regainer-v2 checkpoint
48765ea
raw
history blame
457 Bytes
[[datasets]]
resolution = 512
min_bucket_reso = 256
max_bucket_reso = 1024
caption_dropout_rate = 0
caption_tag_dropout_rate = 0
caption_dropout_every_n_epochs = 0
flip_aug = true
color_aug = false
[[datasets.subsets]]
image_dir = "/content/dreambooth/train_data"
class_tokens = "mksks style"
num_repeats = 2
[general]
enable_bucket = true
caption_extension = ".txt"
shuffle_caption = true
keep_tokens = 0
bucket_reso_steps = 64
bucket_no_upscale = false