Trained for 5 epochs and 180000 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 (14) hide show

README.md +1 -1
optimizer.bin +1 -1
random_states_0.pkl +1 -1
scheduler.bin +1 -1
training_state-anatomy.json +0 -0
training_state-bg20k-1024.json +0 -0
training_state-dalle3.json +2 -2
training_state-midjourney-v6-520k-raw.json +2 -2
training_state-mj-60.json +0 -0
training_state-nijijourney-v6-520k-raw.json +2 -2
training_state-sfwbooru.json +0 -0
training_state-text-1mp.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: 5
-- Training steps: 179000
 - Learning rate: 1e-06
 - Effective batch size: 64
   - Micro-batch size: 8

 ## Training settings
 - Training epochs: 5
+- Training steps: 180000
 - 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:3f67e511d1991b81d1855a6716e7d68343cb5810b398dfc3ad71f78a7f65b189
 size 5451415117

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

random_states_0.pkl CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:08994cee17af3b58aa297bc981c899a0da94a3438919086d06d5b74366934f93
 size 16100

scheduler.bin CHANGED Viewed

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

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

training_state-anatomy.json CHANGED Viewed

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

training_state-bg20k-1024.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:e572f11aa6d7eb63b74784b697f193aafaccf8185fe4afc2b1d23afd2cb68fa3
-size 15326600

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

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

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:10c5f76800d6abe27741d16d41260b991a0aa0b894c22cd38729457a94ec746c
+size 2733018

training_state-mj-60.json CHANGED Viewed

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

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

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:49603423c864c8dab58cfa2193f10f2b0d7754951b511c00f75eefb3bb3311da
+size 2912210

training_state-sfwbooru.json CHANGED Viewed

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

training_state-text-1mp.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": ~~179000~~, "epoch_step": ~~179005~~, "epoch": 6, "exhausted_backends": ["signs", "guys", "nijijourney", "bookcovers", "celebrities", "normalnudes", "sports", "moviecollection", "gay", "experimental", "yoga", "cinemamix-1mp", "ethnic", "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": 4, "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": 180000, "epoch_step": 180005, "epoch": 6, "exhausted_backends": ["signs", "guys", "nijijourney", "bookcovers", "celebrities", "normalnudes", "sports", "moviecollection", "gay", "experimental", "yoga", "cinemamix-1mp", "ethnic", "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": 5, "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 CHANGED Viewed

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

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