Trained for 2 epochs and 81000 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 +2 -2
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: 2
-- Training steps: 80000
 - Learning rate: 1e-06
 - Effective batch size: 64
  - Micro-batch size: 8

 ## Training settings
 - Training epochs: 2
+- Training steps: 81000
 - 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:4a4400f3255799925b5ebda3045d745f5cd87f9564a5740dc77a7d076dcb477e
 size 5451415117

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

random_states_0.pkl CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:6ee5940d80363b8855419120c1691bf1dd7f5f7a0a741ea957e6fcf871795cde
+size 16036

scheduler.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:d1801935c7d3a4c3611846d4634bdee075f7fbdc236b7d7e8dd217c0fa8c76f3
 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:19ba213904dd8c3106d6f3a89997e7347f3f21c0733fe8fda6c48ad6562f2ccd
-size 15502654

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

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

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:52dd9eddc247f5b9e9827c47fb6ae67d3584c2e57500c988064c093205ae1a2b
+size 2847498

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:0cb103aeb974eefa88eb46d4c3c74a5ac152854b1e3d77c66e1e7a594efb5e24
-size 2924930

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

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": ~~80000~~, "epoch_step": ~~80002~~, "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"], "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": 2, "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}}

+ {"global_step": 81000, "epoch_step": 81002, "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"], "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": 2, "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:e206eaf9780e5f60d1ffc324cdb22ecae92f7966a0b8fa013f1675a973bd4188
 size 1816969728

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