PseudoTerminal X commited on
Commit
b42f3ea
1 Parent(s): 7e73171

Trained for 13 epochs and 23000 steps.

Browse files

Trained with datasets ['text-embeds-sdxl-nofilter', 'photo-concept-bucket']
Learning rate 4e-07, batch size 32, and 2 gradient accumulation steps.
Used DDPM noise scheduler for training with v_prediction prediction type and rescaled_betas_zero_snr=True
Using 'trailing' timestep spacing.
Base model: ptx0/terminus-xl-velocity-v2
VAE: madebyollin/sdxl-vae-fp16-fix

README.md CHANGED
@@ -1003,8 +1003,8 @@ You may reuse the base model text encoder for inference.
1003
 
1004
  ## Training settings
1005
 
1006
- - Training epochs: 12
1007
- - Training steps: 22000
1008
  - Learning rate: 4e-07
1009
  - Effective batch size: 512
1010
  - Micro-batch size: 32
 
1003
 
1004
  ## Training settings
1005
 
1006
+ - Training epochs: 13
1007
+ - Training steps: 23000
1008
  - Learning rate: 4e-07
1009
  - Effective batch size: 512
1010
  - Micro-batch size: 32
optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c3576f2e8e8bd66a43a4439ed3d6ed46001d0603aee2646632224702ae2b4023
3
  size 15406336826
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:00df4bf15c44d537207d71a1575b6d3517a37080f2e5f6fd7e2b3544a92ebf4a
3
  size 15406336826
random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:31b25609f77e9a2b895f0971a47c63bd003b5ab22b3ddf7ca070bc64cec1d9d8
3
  size 16100
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:94fb71a0feb3c4002573811a7310e3370b3a17974f46ee77c30522246f010092
3
  size 16100
scheduler.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4e4454174ebab232fc0d463de2e9762286993a7b20c65f0bf1c7a237879eba76
3
  size 1000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:beadde2399f3e9409f4f74fb4af57f3b5fe6793b71901518cfd009359e05e7d6
3
  size 1000
training_state-photo-concept-bucket.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f2dd7b23b5eebcf11d86f3a17cb761fe62dce45881f23f1f92c02f2b924bfb44
3
- size 6218169
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1db1e22422b59e69610018b0eda0f8dad09a0bb7c6a452c54e62d08ee178596
3
+ size 5826937
training_state.json CHANGED
@@ -1 +1 @@
1
- {"global_step": 22000, "epoch_step": 421, "epoch": 13, "exhausted_backends": [], "repeats": {"celebrities": 0, "pixel-art": 0, "movieposters": 0, "moviecollection": 0, "training-test": 0, "dalle3": 0, "photo-concept-bucket": 0}}
 
1
+ {"global_step": 23000, "epoch_step": 332, "epoch": 14, "exhausted_backends": [], "repeats": {"celebrities": 0, "pixel-art": 0, "movieposters": 0, "moviecollection": 0, "training-test": 0, "dalle3": 0, "photo-concept-bucket": 0}}
unet/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
- "_diffusers_version": "0.22.0.dev0",
4
- "_name_or_path": "/notebooks/datasets/models/checkpoint-17400",
5
  "act_fn": "silu",
6
  "addition_embed_type": "text_time",
7
  "addition_embed_type_num_heads": 64,
@@ -51,6 +51,7 @@
51
  "resnet_out_scale_factor": 1.0,
52
  "resnet_skip_time_act": false,
53
  "resnet_time_scale_shift": "default",
 
54
  "sample_size": 128,
55
  "time_cond_proj_dim": null,
56
  "time_embedding_act_fn": null,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.29.0.dev0",
4
+ "_name_or_path": "/home/user/training/lite-models/checkpoint-19000",
5
  "act_fn": "silu",
6
  "addition_embed_type": "text_time",
7
  "addition_embed_type_num_heads": 64,
 
51
  "resnet_out_scale_factor": 1.0,
52
  "resnet_skip_time_act": false,
53
  "resnet_time_scale_shift": "default",
54
+ "reverse_transformer_layers_per_block": null,
55
  "sample_size": 128,
56
  "time_cond_proj_dim": null,
57
  "time_embedding_act_fn": null,
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0ada4a9077a3d8b41dd43564a0b1587c976453f86678242a5c7e1f8d412995ed
3
  size 5135151440
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e76dd7e34b553ed429cdd5477993fdd9b8aa5ff9e998a52f04ba3f20eb581d89
3
  size 5135151440