PseudoTerminal X commited on
Commit
306688f
·
verified ·
1 Parent(s): 36a8fc0

Trained for 4 epochs and 5500 steps.

Browse files

Trained with datasets ['text-embeds-pixart-nofilter', 'photo-concept-bucket']
Learning rate 1e-06, batch size 32, and 2 gradient accumulation steps.
Used DDPM noise scheduler for training with epsilon prediction type and rescaled_betas_zero_snr=False
Using 'trailing' timestep spacing.
Base model: PixArt-alpha/PixArt-Sigma-XL-2-1024-MS
VAE: madebyollin/sdxl-vae-fp16-fix

README.md CHANGED
@@ -1005,8 +1005,8 @@ You may reuse the base model text encoder for inference.
1005
 
1006
  ## Training settings
1007
 
1008
- - Training epochs: 3
1009
- - Training steps: 5000
1010
  - Learning rate: 1e-06
1011
  - Effective batch size: 512
1012
  - Micro-batch size: 32
@@ -1024,7 +1024,7 @@ You may reuse the base model text encoder for inference.
1024
  ### photo-concept-bucket
1025
  - Repeats: 0
1026
  - Total number of images: ~550912
1027
- - Total number of aspect buckets: 1
1028
  - Resolution: 1.0 megapixels
1029
  - Cropped: False
1030
  - Crop style: None
 
1005
 
1006
  ## Training settings
1007
 
1008
+ - Training epochs: 4
1009
+ - Training steps: 5500
1010
  - Learning rate: 1e-06
1011
  - Effective batch size: 512
1012
  - Micro-batch size: 32
 
1024
  ### photo-concept-bucket
1025
  - Repeats: 0
1026
  - Total number of images: ~550912
1027
+ - Total number of aspect buckets: 2
1028
  - Resolution: 1.0 megapixels
1029
  - Cropped: False
1030
  - Crop style: None
optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7cca25ecf2e85a5cb3fbbd1d3ed3766d683c43b4b08db16f665d85b582659151
3
  size 3665677155
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf4ec21d442bb85461467ac52de2b4a65fda8fd8acc9ae87c77fe0bd880ddafc
3
  size 3665677155
random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:636846c25af7f7b96e1b861d9623a517145833d2a60a8d950172e9c02c057731
3
  size 16100
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bfd0c7a7b81f87f1866bf49f48b788d38c4b202d5ec2e9b5f4c81dfd9cefe36e
3
  size 16100
scheduler.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5559c695ac56f52fc3d71f9c30fb72334bb2da74f4e0e82e589d309667a6ca0f
3
  size 1000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8455dbf5b441fa4ecbf748064df8779303204e3a230a7228ca16030675e12075
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:bf5b21dd035ce7eeda64e3a1449dc883e10afb635f856ae7f8472df0c043d768
3
- size 8565369
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61641d895a4535c9f20dd0971a87e290fb5bf5eb054a58551703846261580866
3
+ size 6035795
training_state.json CHANGED
@@ -1 +1 @@
1
- {"global_step": 5000, "epoch_step": 967, "epoch": 4, "exhausted_backends": [], "repeats": {"photo-concept-bucket": 0}}
 
1
+ {"global_step": 5500, "epoch_step": 391, "epoch": 5, "exhausted_backends": [], "repeats": {"photo-concept-bucket": 0}}
transformer/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:648dcad08b1a94f9d35a9e2321f9b65d6b2b26205af9cbc2153437a9e57d795f
3
  size 1221780352
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4005e5da302326631c2a860d03b01f1b93c2211ea71ed366759d2f7388483a87
3
  size 1221780352