Trained for 1 epochs and 41000 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: 1
-- Training steps: 40000
+- Training steps: 41000
 - 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:07cc6bdc929f4e847fd98dd635456ac736d39a8777b3f8468a7ac28575d19935
+oid sha256:1e790b53d163acc0de7770b22972f06fc80e681fedb731d761866c2833f6bf32
 size 5451415117

random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fa5301a02d8746e24ad12af4c2e433ab3af31114e4ceec18e3d050f1b79de4ba
-size 16036
+oid sha256:e9ea867ce29b2683e348f1669dd91c4b33f373cfcb11d1bfc228783dbdfd33ae
+size 16100

scheduler.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a1462ffbb4637bd16dc20793bc5cdc4195eb636c6d8154af043828d73f3ce49f
+oid sha256:2d30104d57545a8f6edcd39f9e14f4b6190ee10a3f20d261d6b4b50bc352e0af
 size 1000

training_state-dalle3.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:da92e2f9ff346c2dc4cfa6c68e59f2d8deada27c8c08dc4610f13bb6492b571c
-size 11756262
+oid sha256:9613dfdbb7bac8ead03045c0091dc940c60af48e490f1380a13e9d141dd5342f
+size 11817869

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3f7b1c032d11cd715af5e7e85675b7cb95f3af560f8977463cd6c03ba4cf0fc6
-size 2486250
+oid sha256:d315aaa0d833122904519e28463e0a0ef9223af12d981194195538e0e5774ffa
+size 2539674

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d5c6a4f0cbf12828008e4a1bd16b56067ae02c56585b64cccf7404f35f505d6a
-size 2618378
+oid sha256:6edd25d9109b5a7230cbd9b1def07bd12f9733eb466c8175feedb08a64c3486d
+size 2666714

training_state.json

@@ -1 +1 @@
-{"global_step": 40000, "epoch_step": 40001, "epoch": 2, "exhausted_backends": ["guys", "signs", "nijijourney", "normalnudes", "bookcovers", "celebrities", "moviecollection", "gay", "sports", "experimental", "ethnic", "cinemamix-1mp", "yoga", "architecture", "shutterstock", "nsfw-1024"], "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": 1, "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}}
+{"global_step": 41000, "epoch_step": 41001, "epoch": 2, "exhausted_backends": ["guys", "signs", "nijijourney", "normalnudes", "bookcovers", "celebrities", "moviecollection", "gay", "sports", "experimental", "ethnic", "cinemamix-1mp", "yoga", "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": 1, "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}}

transformer/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:96ba319f73b480b9095f7e0a182360dee08b86845fe79760f42a81c4cd7e8880
+oid sha256:9b1353f206ec3f62396b7db10dbe47e5f3485360c12ea16e18cf111b46a009ea
 size 1816969728