[model_arguments] | |
v2 = true | |
v_parameterization = true | |
pretrained_model_name_or_path = "/content/pretrained_model/Replicant-V3.0_fp16.safetensors" | |
[optimizer_arguments] | |
min_snr_gamma = 5 | |
optimizer_type = "AdamW8bit" | |
learning_rate = 2e-6 | |
max_grad_norm = 1.0 | |
lr_scheduler = "constant" | |
lr_warmup_steps = 0 | |
[dataset_arguments] | |
cache_latens = false | |
debug_dataset = false | |
vae_batch_size = 1 | |
[training_arguments] | |
output_dir = "/content/dreambooth/output" | |
output_name = "reserval-v3-base_model" | |
save_precision = "fp16" | |
save_n_epoch_ratio = 1 | |
save_state = true | |
resume = "/content/dreambooth/reserval-v2-state" | |
train_batch_size = 2 | |
max_token_length = 225 | |
mem_eff_attn = false | |
xformers = true | |
max_train_steps = 7000 | |
max_data_loader_n_workers = 8 | |
persistent_data_loader_workers = true | |
gradient_checkpointing = false | |
gradient_accumulation_steps = 1 | |
mixed_precision = "fp16" | |
logging_dir = "/content/dreambooth/logs" | |
log_prefix = "reserval-v3-base_model" | |
[sample_prompt_arguments] | |
sample_every_n_steps = 100 | |
sample_sampler = "ddim" | |
[dreambooth_arguments] | |
prior_loss_weight = 1.0 | |
[saving_arguments] | |
save_model_as = "safetensors" | |