Trained for 5 epochs and 177000 steps.

Trained with datasets ['text-embeds', 'text-embeds-extra', 'image-embeds', 'sports', 'mj-60', 'id-75k', 'celebrities', 'normalnudes', 'guys', 'signs', 'dalle3', 'sfwbooru', 'moviecollection', 'bookcovers', 'nijijourney', 'experimental', 'ethnic', 'gay', 'architecture', 'shutterstock', 'midjourney-v6-520k-raw', 'nijijourney-v6-520k-raw', 'cinemamix-1mp', 'nsfw-1024', 'anatomy', 'bg20k-1024', 'yoga', 'photo-aesthetics', 'text-1mp']
Learning rate 1e-06, batch size 8, 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-v0.7-stage1
VAE: ptx0/pixart-900m-1024-ft-v0.7-stage1

README.md

@@ -552,7 +552,7 @@ You may reuse the base model text encoder for inference.
 ## Training settings
 
 - Training epochs: 5
-- Training steps: 176000
+- Training steps: 177000
 - Learning rate: 1e-06
 - Effective batch size: 64
  - Micro-batch size: 8

optimizer.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:407f1e61bb84f593e83ec9125cd8114a71ad38fbaae5e8e580bffabb760e4266
+oid sha256:c491d9d86d1063bb0fd6de8b0f36cdb06538e0274fafce5b482609632c663423
 size 5451415117

random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4025b0d7f07bd0c1e3df76ad1145e2ede88ddfbba24ecfacc2e33383ca5d3745
+oid sha256:327edfd83a75d19247b0faf402d5cad9870e60ca595d7f72328fb02501e4e4f3
 size 16100

scheduler.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8b99b225cf3f84aecae377ca5f322b640f97b1ee5917629fe270d104c302d19a
+oid sha256:ef3cd15c3da91d6afc2c69a28854d7a404d871ba8447738bc189c2126030e37f
 size 1000

training_state-dalle3.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6b045b9dc3b1f5526fb2627da6caeaf68abd7748454dbf7a40a2292f8fdbc937
-size 15113520
+oid sha256:a76c3a6b612e723384a3d132e6448ee3b609461f244a48ac1f2778080fb85d87
+size 15180150

training_state-midjourney-v6-520k-raw.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:47a18d43cc7fccb3d9d5a97f2ad8016b2da4dc1150bc7ab1c3f750e68c64cae2
-size 2523562
+oid sha256:c3cf53cadf9812768ea7437908e13e3cb86cf8df4ff80885265ed57e896ad3f0
+size 2583770

training_state-nijijourney-v6-520k-raw.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a03ea72d29304e6badc6ff32576b6be4da8d3287416046f924d2d1cada62eefc
-size 2670106
+oid sha256:30d25a2f2e36f32df034eb1a32724ef85fb21c4405ef3370d3e97071c8496193
+size 2720562

training_state.json

@@ -1 +1 @@
-{"global_step": 176000, "epoch_step": 176005, "epoch": 6, "exhausted_backends": ["signs", "guys", "nijijourney", "bookcovers", "celebrities", "normalnudes", "sports", "moviecollection", "gay", "experimental", "yoga", "cinemamix-1mp", "ethnic", "architecture", "shutterstock", "nsfw-1024", "photo-aesthetics"], "repeats": {"nijijourney": 0, "signs": 0, "guys": 0, "celebrities": 0, "bookcovers": 0, "normalnudes": 0, "moviecollection": 0, "sports": 0, "gay": 0, "experimental": 0, "yoga": 0, "ethnic": 0, "architecture": 0, "cinemamix-1mp": 0, "photo-aesthetics": 0, "anatomy": 0, "text-1mp": 2, "nsfw-1024": 0, "shutterstock": 0, "id-75k": 0, "bg20k-1024": 0, "mj-60": 0, "sfwbooru": 0, "midjourney-v6-520k-raw": 0, "nijijourney-v6-520k-raw": 0, "dalle3": 0}}
+{"global_step": 177000, "epoch_step": 177005, "epoch": 6, "exhausted_backends": ["signs", "guys", "nijijourney", "bookcovers", "celebrities", "normalnudes", "sports", "moviecollection", "gay", "experimental", "yoga", "cinemamix-1mp", "ethnic", "architecture", "shutterstock", "nsfw-1024", "photo-aesthetics", "id-75k"], "repeats": {"nijijourney": 0, "signs": 0, "guys": 0, "celebrities": 0, "bookcovers": 0, "normalnudes": 0, "moviecollection": 0, "sports": 0, "gay": 0, "experimental": 0, "yoga": 0, "ethnic": 0, "architecture": 0, "cinemamix-1mp": 0, "photo-aesthetics": 0, "anatomy": 1, "text-1mp": 3, "nsfw-1024": 0, "shutterstock": 0, "id-75k": 0, "bg20k-1024": 0, "mj-60": 0, "sfwbooru": 0, "midjourney-v6-520k-raw": 0, "nijijourney-v6-520k-raw": 0, "dalle3": 0}}

transformer/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7e42f9c9423df2215d21c1d127f8c7a3d80c8bbd27515b93c93eae87ec223edb
+oid sha256:85a26059f47b66e586ac39923d664a05c759751dfbebe9b15bf8d5995e39a514
 size 1816969728