Trained for 6 epochs and 205000 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: 204000
+- Training steps: 205000
 - 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:b40389af2df2dd8bbfba212c32c42660dae0f01de0ac4d0ba1329f55abde43bb
+oid sha256:4ad94b5acb1c8900508c61daf2ed67fe1371f7adf1f0308e16fb4e378b5fdf1d
 size 5451415117

random_states_0.pkl

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

scheduler.bin

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

training_state-dalle3.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:108e1643f9606a58cca5e363cbc1e9a391d1e54a90a359649099577dbfc577f9
-size 8182616
+oid sha256:33e3ece871848dbf2faf73c1725659255e56a16e73097d6e1f32273665210e45
+size 8227525

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f7dc69279d7a561b7f3365f576fb563e27365ed26dbf65b3c8046b0fd10c76de
-size 2355234
+oid sha256:6ca4b487f74588e7971e6d9f6e4e1c664d01f96a9dc719c99cb5900ebcd5f526
+size 2395938

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0412ef43d9781c3384b6e06c4ca1e850bea9812cfacf0334f09304ae02c6f251
-size 2499658
+oid sha256:5b60a20743c1a3d58125f08ef2865aa32048cae30c676dfd11413595e652fc9a
+size 2540362

training_state.json

@@ -1 +1 @@
-{"global_step": 204000, "epoch_step": 204006, "epoch": 7, "exhausted_backends": ["nijijourney", "guys", "signs", "normalnudes", "bookcovers", "celebrities", "sports", "moviecollection", "gay", "yoga", "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, "dalle3": 0}}
+{"global_step": 205000, "epoch_step": 205006, "epoch": 7, "exhausted_backends": ["nijijourney", "guys", "signs", "normalnudes", "bookcovers", "celebrities", "sports", "moviecollection", "gay", "yoga", "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, "dalle3": 0}}

transformer/diffusion_pytorch_model.safetensors

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