args.json

{
  "shuffle_per_epoch": false,
  "attention": "xformers",
  "model_variant": "base",
  "aspect_mode": "dynamic",
  "aspect_mode_action_preference": "add",
  "use_ema": true,
  "clip_penultimate": false,
  "conditional_dropout": null,
  "disable_cudnn_benchmark": true,
  "use_text_files_as_captions": false,
  "sample_from_batch": 0,
  "stop_text_encoder_training": 999999999999999,
  "use_bucketing": true,
  "regenerate_latent_cache": true,
  "sample_on_training_start": true,
  "add_class_images_to_dataset": false,
  "auto_balance_concept_datasets": false,
  "sample_aspect_ratios": false,
  "dataset_repeats": 1,
  "save_every_n_epoch": 5,
  "pretrained_model_name_or_path": "mekrod/vmp1",
  "pretrained_vae_name_or_path": "input_vae_model/VAE",
  "tokenizer_name": null,
  "instance_data_dir": null,
  "class_data_dir": null,
  "instance_prompt": null,
  "class_prompt": null,
  "save_sample_prompt": null,
  "n_save_sample": 1,
  "sample_height": 512,
  "sample_width": 512,
  "save_guidance_scale": 7.5,
  "save_infer_steps": 30,
  "with_prior_preservation": false,
  "prior_loss_weight": 1.0,
  "num_class_images": 206,
  "output_dir": "output/vmp1e",
  "seed": 3534554,
  "resolution": 512,
  "center_crop": false,
  "train_text_encoder": false,
  "train_batch_size": 1,
  "sample_batch_size": 4,
  "num_train_epochs": 20,
  "max_train_steps": 1140,
  "gradient_accumulation_steps": 1,
  "gradient_checkpointing": true,
  "learning_rate": 1e-06,
  "scale_lr": false,
  "lr_scheduler": "constant",
  "lr_warmup_steps": 0,
  "use_8bit_adam": true,
  "adam_beta1": 0.9,
  "adam_beta2": 0.999,
  "adam_weight_decay": 0.01,
  "adam_epsilon": 1e-08,
  "max_grad_norm": 1.0,
  "push_to_hub": false,
  "hub_token": null,
  "hub_model_id": null,
  "logging_dir": "logs",
  "log_interval": 10,
  "sample_step_interval": 500,
  "mixed_precision": "fp16",
  "local_rank": -1,
  "concepts_list": [
    {
      "instance_prompt": "Magali Villeneuve",
      "instance_data_dir": "datasets/Magali Villeneuve",
      "class_prompt": "",
      "class_data_dir": "",
      "do_not_balance": 0,
      "use_sub_dirs": 0
    },
    {
      "instance_prompt": "Bo Chen",
      "instance_data_dir": "datasets/Bo Chen",
      "class_prompt": "",
      "class_data_dir": "",
      "do_not_balance": 0,
      "use_sub_dirs": 0
    },
    {
      "instance_prompt": "Chengwei Pan",
      "instance_data_dir": "datasets/Chengwei Pan",
      "class_prompt": "",
      "class_data_dir": "",
      "do_not_balance": 0,
      "use_sub_dirs": 0
    }
  ],
  "save_sample_controlled_seed": null,
  "detect_full_drive": true,
  "send_telegram_updates": false,
  "telegram_chat_id": "0",
  "telegram_token": "0",
  "use_deepspeed_adam": false,
  "append_sample_controlled_seed_action": null,
  "add_sample_prompt": null,
  "use_image_names_as_captions": true,
  "mask_prompts": null,
  "batch_tokens": null
}