Trained for 2 epochs and 91000 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

Files changed (11) hide show

README.md +1 -1
optimizer.bin +1 -1
random_states_0.pkl +2 -2
scheduler.bin +1 -1
training_state-anatomy.json +0 -0
training_state-dalle3.json +2 -2
training_state-midjourney-v6-520k-raw.json +2 -2
training_state-nijijourney-v6-520k-raw.json +2 -2
training_state-sfwbooru.json +0 -0
training_state.json +1 -1
transformer/diffusion_pytorch_model.safetensors +1 -1

README.md CHANGED Viewed

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

 ## Training settings
 - Training epochs: 2
+- Training steps: 91000
 - Learning rate: 1e-06
 - Effective batch size: 64
  - Micro-batch size: 8

optimizer.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:0bac8eed1baf19d7a68c2492a81743eac928509f31ef13cf7786d8facd4dbac5
 size 5451415117

random_states_0.pkl CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:a762ba1e56c27c60ce081607fc9dc12e2391e5279dcc22adfb0ee81f6ed15d3a
+size 16164

scheduler.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:510a52bf61661a1b4b2348533967169e07d20dfc47616f80e985fe6db038a96b
 size 1000

training_state-anatomy.json CHANGED Viewed

The diff for this file is too large to render. See raw diff

training_state-dalle3.json CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:748fc2cdc911f3dc6f9ffd62b7946046a0a3b3c2aeacae2e57ea08465938c894
+size 16553437

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

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:939e6f019aaf5d32752625f7a3b908428186be1e6cabbec1cfe43c58fc248ff9
+size 3642074

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

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:d1d531ac3d887f18f19e4f73e7080b6b915a67b00a6ffb42b48da2b0aad1ae00
+size 3754698

training_state-sfwbooru.json CHANGED Viewed

The diff for this file is too large to render. See raw diff

training_state.json CHANGED Viewed

@@ -1 +1 @@

- {"global_step": ~~90000~~, "epoch_step": ~~90002~~, "epoch": 3, "exhausted_backends": ["guys", "signs", "nijijourney", "bookcovers", "normalnudes", "celebrities", "gay", "sports", "moviecollection", "experimental", "ethnic", "cinemamix-1mp", "architecture", "yoga", "nsfw-1024", "shutterstock", "photo-aesthetics", "id-75k", "text-1mp", "bg20k-1024", "mj-60"], "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": 5, "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": 91000, "epoch_step": 91002, "epoch": 3, "exhausted_backends": ["guys", "signs", "nijijourney", "bookcovers", "normalnudes", "celebrities", "gay", "sports", "moviecollection", "experimental", "ethnic", "cinemamix-1mp", "architecture", "yoga", "nsfw-1024", "shutterstock", "photo-aesthetics", "id-75k", "text-1mp", "bg20k-1024", "mj-60", "anatomy"], "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}}

transformer/diffusion_pytorch_model.safetensors CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:d0474694066fa9fbee77faa9d748738be63d6696c4a2dd779c9e4fa66ac25652
 size 1816969728