training_config.json

{
  "model_path": "stabilityai/stable-diffusion-xl-base-1.0",
  "output_dir": "./output",
  "learning_rate": 1e-06,
  "num_epochs": 1,
  "batch_size": 4,
  "gradient_accumulation_steps": 1,
  "max_grad_norm": 1.0,
  "warmup_steps": 0,
  "data_dir": "/workspace/pseudo-camera-10k",
  "cache_dir": "latents_cache",
  "num_inference_steps": 28,
  "use_adafactor": false,
  "enable_compile": false,
  "compile_mode": "default",
  "gradient_checkpointing": false,
  "ema_decay": 0.9999,
  "use_ema": false,
  "finetune_vae": false,
  "vae_learning_rate": 1e-06,
  "vae_train_freq": 10,
  "min_tag_weight": 0.1,
  "max_tag_weight": 3.0,
  "use_tag_weighting": false,
  "save_checkpoints": false,
  "resume_from_checkpoint": null,
  "use_wandb": true,
  "wandb_project": "nai3",
  "wandb_run_name": "highsigma",
  "logging_steps": 10,
  "save_epochs": 1,
  "validation_frequency": 1,
  "skip_validation": true,
  "validation_prompts": [
    "a detailed portrait of a girl",
    "completely black",
    "a red ball on top of a blue cube, both infront of a green triangle"
  ],
  "push_to_hub": false,
  "hub_model_id": null,
  "hub_private": false
}