PseudoTerminal X commited on
Commit
168a427
1 Parent(s): d014a12

Trained for 9 epochs and 23500 steps.

Browse files

Trained with datasets ['text-embeds', 'shutterstock', 'nijijourney', 'bg20k-1024', 'photo-aesthetics', 'text-1mp', 'cinemamix-1mp']
Learning rate 1e-06, batch size 16, and 1 gradient accumulation steps.
Used DDPM noise scheduler for training with epsilon prediction type and rescaled_betas_zero_snr=False
Using 'linspace' timestep spacing.
Base model: terminusresearch/pixart-900m-1024-ft-v0.6
VAE: madebyollin/sdxl-vae-fp16-fix

README.md CHANGED
@@ -552,7 +552,7 @@ You may reuse the base model text encoder for inference.
552
  ## Training settings
553
 
554
  - Training epochs: 9
555
- - Training steps: 23000
556
  - Learning rate: 1e-06
557
  - Effective batch size: 16
558
  - Micro-batch size: 16
 
552
  ## Training settings
553
 
554
  - Training epochs: 9
555
+ - Training steps: 23500
556
  - Learning rate: 1e-06
557
  - Effective batch size: 16
558
  - Micro-batch size: 16
optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4d1caf1f241c3ed7f24190969a1187bdcae030a6a6d6d0ef93ff5a1f6cbd92aa
3
  size 5451415117
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4267f94b79672e5a1fff1ca6609a7974afbef37a1aa50f7976391d398c06ad0e
3
  size 5451415117
random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9c545ec6d6d4e4d3fc89f346151e78254d046d5c9de88f02f3a8aaecbc0e7fe7
3
- size 14344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:00f8697d0f0e227ce501e0f11eadfc8acb01270462325032ae6b0ced6aeae8ec
3
+ size 14408
scheduler.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1df5fbd5d175197bd243cdc955d2c390cabbe37f05c116a415645acd4b4f62eb
3
  size 1000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:306596970600670792e51d9e437d22fdda6289d035fc20ac68f86e32bb83a7b3
3
  size 1000
training_state-bg20k-1024.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f1fb35b3937c4bf0d8a8722be358c24e9482f7c3b940789c4bb65d9bfa563a88
3
- size 17531873
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:151be3b5aaae2bd879002c7f04622fa163b6d107201a8b79cf7936ebe6f11c7e
3
+ size 17816728
training_state-nijijourney.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-photo-aesthetics.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-shutterstock.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-text-1mp.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state.json CHANGED
@@ -1 +1 @@
1
- {"global_step": 23000, "epoch_step": 23000, "epoch": 10, "exhausted_backends": ["celebrities", "gay", "moviecollection", "ethnic", "cinemamix-1mp"], "repeats": {"movieposters": 20, "celebrities": 0, "normalnudes": 3, "moviecollection": 0, "gay": 0, "ethnic": 0, "nsfw-1024": 0, "experimental": 0, "cinemamix-1mp": 0, "text-1mp": 1}}
 
1
+ {"global_step": 23500, "epoch_step": 23500, "epoch": 10, "exhausted_backends": ["celebrities", "gay", "moviecollection", "ethnic", "cinemamix-1mp"], "repeats": {"movieposters": 20, "celebrities": 0, "normalnudes": 3, "moviecollection": 0, "gay": 0, "ethnic": 0, "nsfw-1024": 0, "experimental": 0, "cinemamix-1mp": 0, "text-1mp": 1}}
transformer/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a3238b3750394ccd5dd8bf549c8716f8485ba8f090cda34b22a1f35894842401
3
  size 1816969728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ad824e2122044d2b9b566e3ff10bebbfd8751522d9c5b536b75a3dcf0f318f0
3
  size 1816969728