Trained for 2 epochs and 72000 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: 2
-- Training steps: 71000
+- Training steps: 72000
 - 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:5e8a199a57a737a5a8b31086f9289f64eb761af66a0773f6f22e3430d999411f
+oid sha256:e6db11f3cfe399c8ada1c512f0d41292c4e6e957df06e3419d9e3c1e3868b378
 size 5451415117

random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d73db15668d432f8ee0d8b381ef750baba7c2cafb61b0e97e645b8ab395aef36
-size 16100
+oid sha256:dafdad75240b8327ce989446cbd5927e635f82c768b9adf7ec8569bf8f5375ff
+size 16164

scheduler.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0c0750030ff7829b44d006c3934ecad6d324a325f534f1a3028193a91b88cdb3
+oid sha256:2780631d76a7ac62e88d183f0024c92589d0dcb5f8ad70a4d099ad271bb36839
 size 1000

training_state-dalle3.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4375f535f6b148ebcd5f3992904d97f6b58ded9633dfe49b483aa9465c7371f4
-size 14969875
+oid sha256:fa7959c28fb2ba417361f94ff015b74032ae21cc18a5b5a37732fa42f01317c0
+size 15011023

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:714c967919f06c86c8d808eff7209add336c62def1d58a65e843aa720588e6be
-size 2350994
+oid sha256:e31d97156189b3ac29e5bd47ca7db4da14da35f1cd2444693caafa1ec1cb0a93
+size 2383218

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:10c34201746c2fbd154d5b4d2b1a1d4fcb4b0d756ff872fba65cbe6e60047671
-size 2494146
+oid sha256:fa077cfa644f6c2bbe264cda395bbda7ca39bce150b8c6f1f98b02c83eeb5438
+size 2536122

training_state.json

@@ -1 +1 @@
-{"global_step": 71000, "epoch_step": 71002, "epoch": 3, "exhausted_backends": ["guys", "signs", "nijijourney", "bookcovers", "normalnudes", "celebrities", "gay", "sports", "moviecollection", "experimental", "ethnic", "cinemamix-1mp", "architecture"], "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": 0, "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": 72000, "epoch_step": 72002, "epoch": 3, "exhausted_backends": ["guys", "signs", "nijijourney", "bookcovers", "normalnudes", "celebrities", "gay", "sports", "moviecollection", "experimental", "ethnic", "cinemamix-1mp", "architecture", "yoga"], "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}}

transformer/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f8a33da0b84f33007fa5f97d3f336f42df62bf5e3054da8c83763d2fd53a4c9a
+oid sha256:a6888210b46d114f5ea043a651557e9bbe05bae386f76e1fbf06cd71f9558146
 size 1816969728