PseudoTerminal X commited on
Commit
a828419
1 Parent(s): a6aba71

Trained for 9 epochs and 29500 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: 29000
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: 29500
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:f0f1a87ce7027c9df4bb157da743de90cbcf81226b08cc9391b95280f8951d01
3
  size 5451415117
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0bc91d8595fc8ae5a0d30ae64329bc26a8d817c981b07e56de9270aaaddf6cd
3
  size 5451415117
random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c71c8545cf72a871d41ab8f17385d91812b6b71f7d635371065d84b43b2dc15a
3
  size 14344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7369191a725b79169ad2afadc3386fc264d7294869d441f1f4ef46c1f2005e7e
3
  size 14344
scheduler.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3efc7413a4d89d5eeb49b1360238abd012dcbf3d569d6276ee9ed92cb768a65d
3
  size 1000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b132c894a077179f14d5c6963bd405481680ec11d9f5529eb0a7909b60f3021
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:d127af0c937cc235f459e7b8aa5b2d83bb12415da25829c2e868e659292ed975
3
- size 23589519
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:872c6f604c61318a699d186fcf23cf15bb90a9ed5942fec4fe019dacd4915dc5
3
+ size 24308499
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": 29000, "epoch_step": 29000, "epoch": 10, "exhausted_backends": ["celebrities", "gay", "moviecollection", "ethnic", "cinemamix-1mp", "shutterstock", "nijijourney", "photo-aesthetics"], "repeats": {"movieposters": 20, "celebrities": 0, "normalnudes": 3, "moviecollection": 0, "gay": 0, "ethnic": 0, "nsfw-1024": 0, "experimental": 0, "cinemamix-1mp": 0, "text-1mp": 3, "shutterstock": 0, "nijijourney": 0, "photo-aesthetics": 0}}
 
1
+ {"global_step": 29500, "epoch_step": 29500, "epoch": 10, "exhausted_backends": ["celebrities", "gay", "moviecollection", "ethnic", "cinemamix-1mp", "shutterstock", "nijijourney", "photo-aesthetics"], "repeats": {"movieposters": 20, "celebrities": 0, "normalnudes": 3, "moviecollection": 0, "gay": 0, "ethnic": 0, "nsfw-1024": 0, "experimental": 0, "cinemamix-1mp": 0, "text-1mp": 4, "shutterstock": 0, "nijijourney": 0, "photo-aesthetics": 0}}
transformer/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d8da14668cbc58f7c040223971538d5b1d31d4fb228080714fbed2e0adf2d238
3
  size 1816969728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ffa879d7af57fcfb3702190e24b3a5e01a9bfb35abdb95151b61d051a118237
3
  size 1816969728