Trained for 6 epochs and 209000 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: 6
-- Training steps: 208000
+- Training steps: 209000
 - 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:fc0781fd2f10e23637bac99058af14597780ec0ace09519d7ca56dbc6b32ce5d
+oid sha256:0936f49a7ce01b615ada7917e1b7f698dafaa6f0ab1ed33ee110efc1a512a37a
 size 5451415117

random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:73d0f23914b4e3bf00f0fdaf94aa53a9dc195662f90e37e80b6a34f705be1bcc
+oid sha256:f315dac669b41e2d5adec175d8883dab540611f9c51def1410e95d14ab06607f
 size 16100

scheduler.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:935f487822f333eb3a83c4eae80bf5b2e0092a3e5b18dc8bc102fb5243d860f3
+oid sha256:a1044147d7a97b0a1bf9cbc29e27d96f9f58c2616f8e4974dc39d2df3ca1f405
 size 1000

training_state-dalle3.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b2eafe9be54b9a4a9ee46f1d26b1ca992021671e27fe1037a022828bfce735ff
-size 8390755
+oid sha256:317aade254ece55f6e429fd49ca17588badf0d4bc18db4165696f2d19e8683ad
+size 8459915

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a7d8a4b27199867d4678c64c99f13a9151bed62373af181476cbe0c16712c563
-size 2534162
+oid sha256:cc8028952be0ea589e6cd5f5a8868c8836b3f47997c5c431abdbca65190547f9
+size 2589706

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:61f7a96f94e1400d1065580fd0f0137df95d680a0a001886f5c71d46a35eba24
-size 2663746
+oid sha256:0c73cda69b9382642cd48e8d47379f73f6eb065a561ec72d7e1b57b2ae60cede
+size 2712930

training_state.json

@@ -1 +1 @@
-{"global_step": 208000, "epoch_step": 208006, "epoch": 7, "exhausted_backends": ["nijijourney", "guys", "signs", "normalnudes", "bookcovers", "celebrities", "sports", "moviecollection", "gay", "yoga", "experimental", "ethnic", "cinemamix-1mp", "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": 209000, "epoch_step": 209006, "epoch": 7, "exhausted_backends": ["nijijourney", "guys", "signs", "normalnudes", "bookcovers", "celebrities", "sports", "moviecollection", "gay", "yoga", "experimental", "ethnic", "cinemamix-1mp", "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:0ea8e5c47abbb73536570ea9114e1c7cb22a22669862b5c5a14cfdf4464ebb2a
+oid sha256:075e66cf37cb029bcca2a86554a884d2ed9832d29b7e1ff58bae06dcebb51e97
 size 1816969728