chenghuangmiao / config_dreambooth-20240826-142529.toml
zbchina's picture
Upload folder using huggingface_hub
b29aca2 verified
raw
history blame
1.31 kB
bucket_no_upscale = true
bucket_reso_steps = 64
cache_latents = true
cache_latents_to_disk = true
clip_skip = 1
dynamo_backend = "no"
epoch = 1
full_fp16 = true
gradient_accumulation_steps = 1
gradient_checkpointing = true
huber_c = 0.1
huber_schedule = "snr"
learning_rate = 1e-5
learning_rate_te1 = 3e-6
logging_dir = "/kaggle/working/outputs/log"
loss_type = "l2"
lr_scheduler = "constant"
lr_scheduler_args = []
lr_scheduler_num_cycles = 1
lr_scheduler_power = 1
max_bucket_reso = 2048
max_data_loader_n_workers = 0
max_timestep = 1000
max_token_length = 75
max_train_steps = 300
min_bucket_reso = 256
mixed_precision = "fp16"
noise_offset_type = "Original"
optimizer_args = [ "scale_parameter=False", "relative_step=False", "warmup_init=False", "weight_decay=0.01",]
optimizer_type = "Adafactor"
output_dir = "/kaggle/temp/models"
output_name = "DreamBooth-chenghuangmiao"
pretrained_model_name_or_path = "stabilityai/stable-diffusion-xl-base-1.0"
prior_loss_weight = 1
resolution = "1024,1024"
sample_prompts = "/kaggle/temp/models/prompt.txt"
sample_sampler = "euler_a"
save_every_n_epochs = 1
save_every_n_steps = 2251
save_model_as = "safetensors"
save_precision = "fp16"
train_batch_size = 2
train_data_dir = "/kaggle/working/outputs/img"
vae = "stabilityai/sdxl-vae"
vae_batch_size = 4
xformers = true