Spaces:
Running
on
Zero
Running
on
Zero
| # GLOBAL STUFF | |
| experiment_id: Ultrapixel_controlnet | |
| checkpoint_path: checkpoint output path | |
| output_path: visual results output path | |
| model_version: 3.6B | |
| dtype: float32 | |
| # # WandB | |
| # wandb_project: StableCascade | |
| # wandb_entity: wandb_username | |
| #module_filters: ['.depthwise', '.mapper', '.attn', '.channelwise' ] | |
| #rank: 32 | |
| # TRAINING PARAMS | |
| lr: 1.0e-4 | |
| batch_size: 12 | |
| #image_size: [1536, 2048, 2560, 3072, 4096] | |
| image_size: [1024, 2048, 2560, 3072, 3584, 3840, 4096, 4608] | |
| #image_size: [ 1024, 1536, 2048, 2560, 3072, 3584, 3840, 4096, 4608] | |
| #image_size: [ 1024, 1280] | |
| multi_aspect_ratio: [1/1, 1/2, 1/3, 2/3, 3/4, 1/5, 2/5, 3/5, 4/5, 1/6, 5/6, 9/16] | |
| grad_accum_steps: 2 | |
| updates: 40000 | |
| backup_every: 5000 | |
| save_every: 256 | |
| warmup_updates: 1 | |
| use_fsdp: True | |
| # ControlNet specific | |
| controlnet_blocks: [0, 4, 8, 12, 51, 55, 59, 63] | |
| controlnet_filter: CannyFilter | |
| controlnet_filter_params: | |
| resize: 224 | |
| # offset_noise: 0.1 | |
| # GDF | |
| adaptive_loss_weight: True | |
| ema_start_iters: 10 | |
| ema_iters: 50 | |
| ema_beta: 0.9 | |
| webdataset_path: path to your training dataset | |
| effnet_checkpoint_path: models/effnet_encoder.safetensors | |
| previewer_checkpoint_path: models/previewer.safetensors | |
| generator_checkpoint_path: models/stage_c_bf16.safetensors | |
| controlnet_checkpoint_path: models/canny.safetensors | |