Trained for 0 epochs and 9000 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

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

 ## Training settings
 - Training epochs: 0
+- Training steps: 9000
 - 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:1ca41a523def9293e6bfa15143482828120858bd53a559aa3b94471cd60df450
 size 5451415117

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

random_states_0.pkl CHANGED Viewed

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

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

scheduler.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:09d20795f45081f74318fb9f922f576394d836bd24d565508e1aeb5538331a62
 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:b923b0c1d24468fae75448b03f53b4ae552f87dafbdc07779ef5cff1ba36b3a1
-size 8492980

 version https://git-lfs.github.com/spec/v1
+oid sha256:30ab6c581b13b05a09bffa50d1ccc0f2beffbaef473264d9be137733960a1f2d
+size 8570865

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

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

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

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:be1a3889d53fd5c11d2d42f80557aa6066d29f1538a8d2ccd0cc65995f1e16b8
-size 2781618

 version https://git-lfs.github.com/spec/v1
+oid sha256:385abc0b41b096a1bfa50a89a3ae4c605b77b8f0fd243c5049026664a003aa15
+size 2826138

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": ~~8000~~, "epoch_step": ~~8000~~, "epoch": 1, "exhausted_backends": ["nijijourney", "signs", "guys", "celebrities", "bookcovers", "normalnudes", "moviecollection", "sports", "gay", "experimental", "yoga", "ethnic", "architecture", "cinemamix-1mp", "photo-aesthetics", "nsfw-1024", "shutterstock", "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": 3, "nsfw-1024": 0, "shutterstock": 0, "id-75k": 0}}

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

transformer/diffusion_pytorch_model.safetensors CHANGED Viewed

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

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