File size: 3,590 Bytes
2684334 461895e 2684334 461895e 2684334 |
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 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 |
{
"training": {
"master_seed": 2,
"cache_minibatch": 8,
"train_minibatch": 4,
"offload_param_count": 17000000000,
"lr": 2e-06,
"weight_decay": 0.0,
"warmup_steps": 5,
"change_layer_every": 100000000000,
"trained_single_blocks": 38,
"trained_double_blocks": 19,
"save_every": 50,
"save_folder": "model_priming_low_lr",
"wandb_project": "optimal transport unlocked",
"wandb_run": "priming",
"wandb_entity": null
},
"inference": {
"inference_every": 20,
"inference_folder": "primed_output",
"steps": 8,
"guidance": 0,
"cfg": 1,
"prompts": [
"a cute cat sat on a mat while receiving a head pat from his owner called Matt",
"baked potato, on the space floating orbiting around the earth"
],
"first_n_steps_wo_cfg": 0,
"image_dim": [
1024,
1024
],
"t5_max_length": 512
},
"extra_inference_config": [
{
"inference_every": 20,
"inference_folder": "schnell_dedistilation",
"steps": 8,
"guidance": 0,
"cfg": 3,
"prompts": [
"a cute cat sat on a mat while receiving a head pat from his owner called Matt",
"baked potato, on the space floating orbiting around the earth"
],
"first_n_steps_wo_cfg": 0,
"image_dim": [
1024,
1024
],
"t5_max_length": 512
},
{
"inference_every": 20,
"inference_folder": "schnell_dedistilation",
"steps": 20,
"guidance": 0,
"cfg": 3,
"prompts": [
"a cute cat sat on a mat while receiving a head pat from his owner called Matt",
"baked potato, on the space floating orbiting around the earth"
],
"first_n_steps_wo_cfg": 0,
"image_dim": [
1024,
1024
],
"t5_max_length": 512
},
{
"inference_every": 20,
"inference_folder": "schnell_dedistilation",
"steps": 20,
"guidance": 0,
"cfg": 7,
"prompts": [
"a cute cat sat on a mat while receiving a head pat from his owner called Matt",
"baked potato, on the space floating orbiting around the earth"
],
"first_n_steps_wo_cfg": 0,
"image_dim": [
1024,
1024
],
"t5_max_length": 512
}
],
"dataloader": {
"batch_size": 1024,
"jsonl_metadata_path": "for_priming.jsonl",
"image_folder_path": "debug_dataset",
"base_resolution": [
512
],
"shuffle_tags": true,
"tag_drop_percentage": 0.0,
"uncond_percentage": 0.0,
"resolution_step": 64,
"num_workers": 0,
"prefetch_factor": null,
"ratio_cutoff": 2.0,
"thread_per_worker": 10
},
"model": {
"chroma_path": "2025-01-25_16-05-41.safetensors",
"vae_path": "FLUX.1-schnell/ae.safetensors",
"t5_path": "FLUX.1-schnell/text_encoder_2",
"t5_config_path": "FLUX.1-schnell/text_encoder_2/config.json",
"t5_tokenizer_path": "FLUX.1-schnell/tokenizer_2",
"t5_to_8bit": true,
"t5_max_length": 512
}
} |