SW-Yoon/avs-canny-controlnet

Description:

{
  "pretrained_model_name_or_path": "black-forest-labs/FLUX.1-dev",
  "pretrained_controlnet_name_or_path": "InstantX/FLUX.1-dev-Controlnet-Canny",
  "pretrained_controlnet_safetensors_name": null,
  "revision": null,
  "variant": null,
  "dataset_dir": "AVS",
  "dataset_py_file": "dataset.py",
  "dataset_config_name": null,
  "cache_dir": null,
  "image_column": "image",
  "max_sequence_length": 512,
  "validation_prompt": null,
  "validation_conditioning_scale": 1.0,
  "num_validation_images": 2,
  "validation_epochs": 50,
  "validation_steps": 500,
  "output_dir": "results/avs/1",
  "seed": null,
  "resolution": 512,
  "num_layers": 2,
  "num_single_layers": 4,
  "control_mode": 0,
  "train_batch_size": 5,
  "num_train_epochs": 5,
  "max_train_steps": null,
  "logging_steps": 100,
  "gpu_logging_steps": 100,
  "checkpointing_steps": 1000,
  "checkpoints_total_limit": null,
  "resume_from_checkpoint": "latest",
  "gradient_accumulation_steps": 2,
  "gradient_checkpointing": false,
  "learning_rate": 2e-05,
  "guidance_scale": 4.0,
  "task": "sketch",
  "save_samples_first_n_steps": 2,
  "scale_lr": false,
  "lr_scheduler": "constant",
  "lr_warmup_steps": 500,
  "lr_num_cycles": 1,
  "lr_power": 1.0,
  "num_workers": 1,
  "weighting_scheme": "logit_normal",
  "logit_mean": 0.0,
  "logit_std": 1.0,
  "mode_scale": 1.29,
  "optimizer": "adamw",
  "use_8bit_adam": true,
  "adam_beta1": 0.9,
  "adam_beta2": 0.999,
  "adam_weight_decay": 0.0001,
  "adam_epsilon": 1e-08,
  "quantize": true,
  "max_grad_norm": 1.0,
  "logging_dir": "logs",
  "allow_tf32": false,
  "mixed_precision": "bf16",
  "validation_path": "validation_images_cc",
  "validation_resolution": 512,
  "validation_batch_size": 2,
  "local_rank": -1
}