|
ae = "/workspace/flux_vae.safetensors" |
|
bucket_no_upscale = true |
|
bucket_reso_steps = 64 |
|
cache_latents_to_disk = true |
|
cache_text_encoder_outputs_to_disk = true |
|
caption_extension = ".txt" |
|
clip_l = "/workspace/clip_l.safetensors" |
|
clip_skip = 1 |
|
discrete_flow_shift = 3.0 |
|
dynamo_backend = "no" |
|
enable_bucket = true |
|
epoch = 10 |
|
full_bf16 = true |
|
gradient_accumulation_steps = 1 |
|
gradient_checkpointing = true |
|
guidance_scale = 3.5 |
|
huber_c = 0.1 |
|
huber_scale = 1 |
|
huber_schedule = "snr" |
|
logging_dir = "/workspace/train/log" |
|
loss_type = "l2" |
|
lr_scheduler = "cosine_with_restarts" |
|
lr_scheduler_args = [] |
|
lr_scheduler_num_cycles = 1 |
|
lr_scheduler_power = 1 |
|
lr_warmup_steps = 0.1 |
|
max_bucket_reso = 2048 |
|
max_data_loader_n_workers = 0 |
|
max_grad_norm = 1 |
|
max_timestep = 1000 |
|
max_train_steps = 150 |
|
metadata_author = "Chris Hsu" |
|
metadata_description = "landscape for Overhead view" |
|
metadata_title = "landscapeOverhead_A6000" |
|
min_bucket_reso = 256 |
|
mixed_precision = "bf16" |
|
model_prediction_type = "sigma_scaled" |
|
network_alpha = 8 |
|
network_args = [ "train_double_block_indices=all", "train_single_block_indices=all",] |
|
network_dim = 4 |
|
network_module = "networks.lora_flux" |
|
noise_offset_type = "Original" |
|
optimizer_args = [] |
|
optimizer_type = "AdamW8bit" |
|
output_dir = "/workspace/train/model" |
|
output_name = "landscapeOverhead_A6000" |
|
pretrained_model_name_or_path = "/workspace/flux1-dev.safetensors" |
|
prior_loss_weight = 1 |
|
resolution = "1024,1024" |
|
sample_every_n_epochs = 1 |
|
sample_prompts = "/workspace/train/model/sample/prompt.txt" |
|
sample_sampler = "euler_a" |
|
save_every_n_epochs = 1 |
|
save_model_as = "safetensors" |
|
save_precision = "bf16" |
|
t5xxl = "/workspace/t5xxl_fp16.safetensors" |
|
t5xxl_max_token_length = 512 |
|
text_encoder_lr = [ 1e-5, 1e-5,] |
|
timestep_sampling = "sigma" |
|
train_batch_size = 2 |
|
train_data_dir = "/workspace/train/img" |
|
unet_lr = 1e-5 |
|
wandb_run_name = "landscapeOverhead_A6000" |
|
xformers = true |
|
|