ptx0 commited on
Commit
04cd503
1 Parent(s): a95ab3a

Trained for 7 epochs and 100000 steps.

Browse files

Trained with datasets ['text-embeds-pixart-filter', 'photo-concept-bucket']
Learning rate 1e-06, batch size 24, 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: ptx0/pixart-900m-1024-ft-large
VAE: madebyollin/sdxl-vae-fp16-fix

README.md CHANGED
@@ -1292,7 +1292,7 @@ You may reuse the base model text encoder for inference.
1292
  ## Training settings
1293
 
1294
  - Training epochs: 7
1295
- - Training steps: 99000
1296
  - Learning rate: 1e-06
1297
  - Effective batch size: 192
1298
  - Micro-batch size: 24
 
1292
  ## Training settings
1293
 
1294
  - Training epochs: 7
1295
+ - Training steps: 100000
1296
  - Learning rate: 1e-06
1297
  - Effective batch size: 192
1298
  - Micro-batch size: 24
optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a3ef8c034fe3a2f33597ab42ec0258930c815b66d3868cc9f819bd7fce8b21ae
3
  size 5451415117
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d9240075af4fda887f942842159a83cd5c0509153f9269612e50d0284ddb0f8
3
  size 5451415117
random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:63b6fab2fa5d5eda067c0e15e07bdafda7e7170a1838b24efaf1706b40a9ff1d
3
  size 16100
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b1e935f84046b9f8460192f432f1cc46de176622681eaa51cc3e960e681b983a
3
  size 16100
scheduler.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2781c871232fc2f4481dac795190a70a0e26ca2a9a93a7b3c66fcfdfa530ace3
3
  size 1000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1df4c5453ccc4a88148dfbebee45f04aea1ed5429edb2ac840aca3128f1b9e98
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:7778c5eba4f5e0256e3cdd70087def9164d19ebbef803554b0dd09b7bf8559f0
3
- size 6556704
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c80d366faeda4c8cce268d92792f36e4384ac03b51b0ca291d19d6e69eeb73f4
3
+ size 8492545
training_state.json CHANGED
@@ -1 +1 @@
1
- {"global_step": 99000, "epoch_step": 1, "epoch": 8, "exhausted_backends": [], "repeats": {"bookcovers": 0, "signs": 0, "normalnudes": 0, "nijijourney": 0, "movieposters": 0, "celebrities": 0, "pixel-art": 0, "propagandaposters": 0, "sports": 0, "moviecollection": 0, "gay": 0, "experimental": 0, "yoga": 0, "ethnic": 0, "cinemamix-1mp": 0, "architecture": 0, "mj-60": 0, "text-1mp": 65, "shutterstock": 0, "nsfw-1024": 0, "photo-aesthetics": 0, "anatomy": 0, "bg20k-1024": 0, "sfwbooru": 0, "midjourney-v6-520k-raw": 0, "nijijourney-v6-520k-raw": 0, "photo-concept-bucket": 0, "dalle3": 0}}
 
1
+ {"global_step": 100000, "epoch_step": 1, "epoch": 8, "exhausted_backends": [], "repeats": {"bookcovers": 0, "signs": 0, "normalnudes": 0, "nijijourney": 0, "movieposters": 0, "celebrities": 0, "pixel-art": 0, "propagandaposters": 0, "sports": 0, "moviecollection": 0, "gay": 0, "experimental": 0, "yoga": 0, "ethnic": 0, "cinemamix-1mp": 0, "architecture": 0, "mj-60": 0, "text-1mp": 65, "shutterstock": 0, "nsfw-1024": 0, "photo-aesthetics": 0, "anatomy": 0, "bg20k-1024": 0, "sfwbooru": 0, "midjourney-v6-520k-raw": 0, "nijijourney-v6-520k-raw": 0, "photo-concept-bucket": 0, "dalle3": 0}}
transformer/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a77717d9f4aa7d86aabfa047dad9df64bace30e3c2f53ef0702b830e23a5fff8
3
  size 1816969728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:705a24337cee8d63e235b18b40344a8775d1d2f6ee6a1b061d92c4075a6d9015
3
  size 1816969728