PseudoTerminal X commited on
Commit
8917b1f
1 Parent(s): ef420be

Trained for 9 epochs and 19500 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: 19000
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: 19500
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:efbbcd07120cd1004dc30391e4c214cfaf6a75357de9d9a0050b8acc247d02c6
3
  size 5451415117
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:620d6048a732cb2a8be9efd3cbf3d3173e314092738b7cb55d699a9c1b61330a
3
  size 5451415117
random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f85315072d2512074ed1e89344b4f15e645342251258d43e927e5ff9d52a9035
3
- size 14408
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:49072f5975971477bf33556cfd7b511e714bbce16d41f950e0d8f9cbfb8e14d5
3
+ size 14344
scheduler.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4b08f92a332633a82924258f32f4f0ffbd630d90db1eb762d572f57b287955c7
3
  size 1000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6803d33ca8df001910c8ae963ffc4af4df0382fa313920e2d980a2a4edd08989
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:7079a83517fcd362a248c0744336f4adcb780eebfaa7d31c7e250ce63dc492ba
3
- size 15350218
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:548ec2aa970ae128454546bb3f7b4b8a153ca98cd922a1fbcf48ca766e4ed082
3
+ size 15646268
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": 19000, "epoch_step": 19000, "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}}
 
1
+ {"global_step": 19500, "epoch_step": 19500, "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}}
transformer/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5f9b628eb57ed5fdd4884d79816e17efe250a3769badef874b5fab9b6b5bae09
3
  size 1816969728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:04b94317e7f37f2b6b9ed88002a655068c46b95b7da01c33a0a33fad54fe025d
3
  size 1816969728