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

 ## Training settings
 - Training epochs: 0
+- Training steps: 13000
 - 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:750cc0002cac492c3eef45851b548584c56c6670f4e4ee269138008d5fcbcc49
 size 5451415117

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

random_states_0.pkl CHANGED Viewed

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

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

scheduler.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:89abbd8f098bd47be661e04b3effa9b2fa5b2e0231d0bc3148a87905693b90d1
 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:d87f018981dff1eb98ae3cceec5f3c73419ff1ac1a86cbf5b5037b7e0300439e
-size 8777805

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

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

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:68a586826e97be9081daa1c4f9c9241bea5d1a1c0eb565d1fb5ba31ba678edfa
+size 2937810

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:292f4847066c498e31eedc34bd00d96a73de231efefd56d36850c7d1776a05a8
-size 2989802

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

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": ~~12000~~, "epoch_step": ~~12000~~, "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": 3, "text-1mp": 5, "nsfw-1024": 0, "shutterstock": 0, "id-75k": 0}}

+ {"global_step": 13000, "epoch_step": 13000, "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", "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": 3, "text-1mp": 0, "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:217f37532507a472da2ab29eab6fdc9679c03ed84c24c4ac93002c179cf02712
 size 1816969728

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