pruebinha_adv / config.yaml
jrochafe's picture
End of training
cf89101 verified
pretrained_vae_model_name_or_path: madebyollin/sdxl-vae-fp16-fix
pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
dataset_name: /home/usr_9106259_ulta_com/illiad/data/identity/pola
instance_prompt: "http woman"
token_abstraction: http
validation_prompt: "selfie photo of http woman wearing nice clothes"
output_dir: pruebinha_adv
caption_column: prompt
mixed_precision: fp16
resolution: 1024
train_batch_size: 3
repeats: 1
rank: 32
report_to: tensorboard
gradient_accumulation_steps: 1
gradient_checkpointing: true
learning_rate: 1.0
text_encoder_lr: 1.0
adam_beta2: 0.99
optimizer: prodigy
train_text_encoder_ti: true
train_text_encoder_ti_frac: 0.5
snr_gamma: 5.0
lr_scheduler: constant
lr_warmup_steps: 0
max_train_steps: 20
checkpointing_steps: 251
seed: 0
with_prior_preservation: true
class_data_dir: /home/usr_9106259_ulta_com/illiad/data/identity/class_images/a_selfie_of_a_woman
class_prompt: "a selfie photo of a woman"
num_class_images: 200
push_to_hub: true
hub_model_id: jrochafe/pruebinha_adv