2024-06-16 09:58:23,174 INFO MainThread:4800 [wandb_setup.py:_flush():76] Current SDK version is 0.17.1 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_setup.py:_flush():76] Configure stats pid to 4800 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_setup.py:_flush():76] Loading settings from /home/Ubuntu/.config/wandb/settings 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_setup.py:_flush():76] Loading settings from /home/Ubuntu/diffusers/examples/dreambooth/wandb/settings 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_setup.py:_flush():76] Loading settings from environment variables: {} 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_setup.py:_flush():76] Applying setup settings: {'_disable_service': False} 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_setup.py:_flush():76] Inferring run settings from compute environment: {'program_relpath': 'examples/dreambooth/train_dreambooth_lora_sd3.py', 'program_abspath': '/home/Ubuntu/diffusers/examples/dreambooth/train_dreambooth_lora_sd3.py', 'program': '/home/Ubuntu/diffusers/examples/dreambooth/train_dreambooth_lora_sd3.py'} 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_setup.py:_flush():76] Applying login settings: {} 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_setup.py:_flush():76] Applying login settings: {'mode': 'offline'} 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_init.py:_log_setup():520] Logging user logs to /home/Ubuntu/diffusers/examples/dreambooth/wandb/offline-run-20240616_095823-nwu6c98b/logs/debug.log 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_init.py:_log_setup():521] Logging internal logs to /home/Ubuntu/diffusers/examples/dreambooth/wandb/offline-run-20240616_095823-nwu6c98b/logs/debug-internal.log 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_init.py:init():560] calling init triggers 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_init.py:init():567] wandb.init called with sweep_config: {} config: {} 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_init.py:init():610] starting backend 2024-06-16 09:58:23,175 INFO MainThread:4800 [wandb_init.py:init():614] setting up manager 2024-06-16 09:58:23,177 INFO MainThread:4800 [backend.py:_multiprocessing_setup():105] multiprocessing start_methods=fork,spawn,forkserver, using: spawn 2024-06-16 09:58:23,178 INFO MainThread:4800 [wandb_init.py:init():622] backend started and connected 2024-06-16 09:58:23,181 INFO MainThread:4800 [wandb_init.py:init():711] updated telemetry 2024-06-16 09:58:23,195 INFO MainThread:4800 [wandb_init.py:init():744] communicating run to backend with 90.0 second timeout 2024-06-16 09:58:23,199 INFO MainThread:4800 [wandb_init.py:init():795] starting run threads in backend 2024-06-16 09:58:26,568 INFO MainThread:4800 [wandb_run.py:_console_start():2380] atexit reg 2024-06-16 09:58:26,568 INFO MainThread:4800 [wandb_run.py:_redirect():2235] redirect: wrap_raw 2024-06-16 09:58:26,568 INFO MainThread:4800 [wandb_run.py:_redirect():2300] Wrapping output streams. 2024-06-16 09:58:26,568 INFO MainThread:4800 [wandb_run.py:_redirect():2325] Redirects installed. 2024-06-16 09:58:26,570 INFO MainThread:4800 [wandb_init.py:init():838] run started, returning control to user process 2024-06-16 09:58:26,571 INFO MainThread:4800 [wandb_run.py:_config_callback():1382] config_cb None None {'pretrained_model_name_or_path': 'stabilityai/stable-diffusion-3-medium-diffusers', 'revision': None, 'variant': None, 'dataset_name': None, 'dataset_config_name': None, 'instance_data_dir': '/home/Ubuntu/dog', 'cache_dir': None, 'image_column': 'image', 'caption_column': None, 'repeats': 1, 'class_data_dir': None, 'instance_prompt': 'a photo of sks dog', 'class_prompt': None, 'validation_prompt': 'A photo of sks dog in a bucket', 'num_validation_images': 4, 'validation_epochs': 25, 'rank': 4, 'with_prior_preservation': False, 'prior_loss_weight': 1.0, 'num_class_images': 100, 'output_dir': 'trained-sd3-lora', 'seed': 0, 'resolution': 512, 'center_crop': False, 'random_flip': False, 'train_batch_size': 1, 'sample_batch_size': 4, 'num_train_epochs': 250, 'max_train_steps': 500, 'checkpointing_steps': 500, 'checkpoints_total_limit': None, 'resume_from_checkpoint': None, 'gradient_accumulation_steps': 4, 'gradient_checkpointing': False, 'learning_rate': 1e-05, 'text_encoder_lr': 5e-06, 'scale_lr': False, 'lr_scheduler': 'constant', 'lr_warmup_steps': 0, 'lr_num_cycles': 1, 'lr_power': 1.0, 'dataloader_num_workers': 0, 'weighting_scheme': 'sigma_sqrt', 'logit_mean': 0.0, 'logit_std': 1.0, 'mode_scale': 1.29, 'optimizer': 'AdamW', 'use_8bit_adam': False, 'adam_beta1': 0.9, 'adam_beta2': 0.999, 'prodigy_beta3': None, 'prodigy_decouple': True, 'adam_weight_decay': 0.0001, 'adam_weight_decay_text_encoder': 0.001, 'adam_epsilon': 1e-08, 'prodigy_use_bias_correction': True, 'prodigy_safeguard_warmup': True, 'max_grad_norm': 1.0, 'push_to_hub': False, 'hub_token': None, 'hub_model_id': None, 'logging_dir': 'logs', 'allow_tf32': False, 'report_to': 'wandb', 'mixed_precision': 'fp16', 'prior_generation_precision': None, 'local_rank': -1} 2024-06-16 10:15:55,967 INFO MainThread:4800 [wandb_run.py:_finish():2109] finishing run dreambooth-sd3-lora/nwu6c98b 2024-06-16 10:15:55,968 INFO MainThread:4800 [wandb_run.py:_atexit_cleanup():2349] got exitcode: 0 2024-06-16 10:15:55,968 INFO MainThread:4800 [wandb_run.py:_restore():2332] restore 2024-06-16 10:15:55,968 INFO MainThread:4800 [wandb_run.py:_restore():2338] restore done 2024-06-16 10:15:57,053 INFO MainThread:4800 [wandb_run.py:_footer_history_summary_info():4008] rendering history 2024-06-16 10:15:57,054 INFO MainThread:4800 [wandb_run.py:_footer_history_summary_info():4040] rendering summary