Trained for 4 epochs and 148000 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: 4
-- Training steps: 147000
 - Learning rate: 1e-06
 - Effective batch size: 64
  - Micro-batch size: 8

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

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

random_states_0.pkl CHANGED Viewed

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

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

scheduler.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:c29cd47f0cfbf61447c7bdffdee2b32ba53ae158a00dacb6ec9ee3e9b37afa4e
 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:ff270d8ea313b070ecaf889738f552398265adbebb832ee4041fd1dbaadf5832
-size 12150411

 version https://git-lfs.github.com/spec/v1
+oid sha256:1d6b0d7406af393ab3b935638677d15360d0c7246791ee007bb83c3b387cff9d
+size 12235701

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

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

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

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:6919bb9e0c36aa6f5548a3215888da265d5e8d2f58e36fb81d3c789dcb13fcd3
-size 2958426

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

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": ~~147000~~, "epoch_step": ~~147004~~, "epoch": 5, "exhausted_backends": ["signs", "bookcovers", "guys", "sports", "normalnudes", "nijijourney", "celebrities", "moviecollection", "gay", "experimental", "ethnic", "cinemamix-1mp", "yoga", "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": 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, "dalle3": 0}}

+ {"global_step": 148000, "epoch_step": 148004, "epoch": 5, "exhausted_backends": ["signs", "bookcovers", "guys", "sports", "normalnudes", "nijijourney", "celebrities", "moviecollection", "gay", "experimental", "ethnic", "cinemamix-1mp", "yoga", "architecture", "shutterstock", "nsfw-1024", "photo-aesthetics", "id-75k", "text-1mp", "bg20k-1024"], "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, "dalle3": 0}}

transformer/diffusion_pytorch_model.safetensors CHANGED Viewed

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

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