Coercer commited on
Commit
a01d294
·
verified ·
1 Parent(s): d368ded

Upload 2 files

Browse files
Training_model_data/TE_Train.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": {
3
+ "type": "sdxl",
4
+ "base_model_path": "SDXLCheckpoint.safetensors",
5
+ "revision": "main",
6
+ "is_full_finetune": true
7
+ },
8
+ "training": {
9
+ "batch_size": 1,
10
+ "learning_rate": 1e-6,
11
+ "seed": 42,
12
+ "max_epochs": 100,
13
+ "mixed_precision": "bf16",
14
+ "gradient_checkpointing": "NONE",
15
+ "layer_offload_fraction": 0.2,
16
+ "save_precision": "bf16",
17
+ "xformers": true,
18
+ "cache_latents": true,
19
+ "train_unet": false,
20
+ "train_text_encoder": true
21
+ },
22
+ "optimizer": {
23
+ "type": "AdamW8bit",
24
+ "params": {
25
+ "weight_decay": 0.01
26
+ },
27
+ "fused_backward_pass": false
28
+ },
29
+ "ema": {
30
+ "active": true,
31
+ "offload": false,
32
+ "decay": 0.999
33
+ },
34
+ "dataset": {
35
+ "concepts": [
36
+ {
37
+ "image_directory": "dataset",
38
+ "caption_extension": ".txt",
39
+ "resolutions": [1024]
40
+ }
41
+ ]
42
+ },
43
+ "output": {
44
+ "directory": "output",
45
+ "save_every_n_epochs": 1,
46
+ "save_name": "sdxl_finetuned"
47
+ }
48
+ }
Training_model_data/Unet_Train.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": {
3
+ "type": "sdxl",
4
+ "base_model_path": "SDXLCheckpoint.safetensors",
5
+ "revision": "main",
6
+ "is_full_finetune": true
7
+ },
8
+ "training": {
9
+ "batch_size": 1,
10
+ "learning_rate": 8e-7,
11
+ "seed": 42,
12
+ "max_epochs": 100,
13
+ "mixed_precision": "bf16",
14
+ "gradient_checkpointing": "CPU_OFFLOADED",
15
+ "layer_offload_fraction": 0.2,
16
+ "save_precision": "bf16",
17
+ "xformers": true,
18
+ "cache_latents": true,
19
+ "train_unet": true,
20
+ "train_text_encoder": false
21
+ },
22
+ "optimizer": {
23
+ "type": "Adafactor",
24
+ "params": {
25
+ "scale_parameter": false,
26
+ "relative_step": false,
27
+ "warmup_init": false
28
+ },
29
+ "fused_backward_pass": true
30
+ },
31
+ "ema": {
32
+ "active": false
33
+ },
34
+ "dataset": {
35
+ "concepts": [
36
+ {
37
+ "image_directory": "dataset",
38
+ "caption_extension": ".txt",
39
+ "resolutions": [1024]
40
+ }
41
+ ]
42
+ },
43
+ "output": {
44
+ "directory": "output",
45
+ "save_every_n_epochs": 1,
46
+ "save_name": "sdxl_finetuned"
47
+ }
48
+ }