thu / config_lora-20250316-041748.toml
housearch's picture
Upload 2 files
ce9ed2f verified
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