Training_model_data/TE_Train.json

{
  "model": {
    "type": "sdxl",
    "base_model_path": "SDXLCheckpoint.safetensors",
    "revision": "main",
    "is_full_finetune": true
  },
  "training": {
    "batch_size": 1,
    "learning_rate": 1e-6,
    "seed": 42,
    "max_epochs": 100,
    "mixed_precision": "bf16",
    "gradient_checkpointing": "NONE",
    "layer_offload_fraction": 0.2,
    "save_precision": "bf16",
    "xformers": true,
    "cache_latents": true,
    "train_unet": false,
    "train_text_encoder": true
  },
  "optimizer": {
    "type": "AdamW8bit",
    "params": {
      "weight_decay": 0.01
    },
    "fused_backward_pass": false
  },
  "ema": {
    "active": true,
    "offload": false,
    "decay": 0.999
  },
  "dataset": {
    "concepts": [
      {
        "image_directory": "dataset",
        "caption_extension": ".txt",
        "resolutions": [1024]
      }
    ]
  },
  "output": {
    "directory": "output",
    "save_every_n_epochs": 1,
    "save_name": "sdxl_finetuned"
  }
}