File size: 1,141 Bytes
c9c23a8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
{
"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
} |