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

README.md

@@ -47,7 +47,7 @@ You may reuse the base model text encoder for inference.
 ## Training settings
 
 - Training epochs: 0
-- Training steps: 6000
+- Training steps: 7000
 - Learning rate: 1e-06
 - Effective batch size: 64
   - Micro-batch size: 8
@@ -80,7 +80,7 @@ You may reuse the base model text encoder for inference.
 - Crop aspect: square
 ### id-75k
 - Repeats: 0
-- Total number of images: ~44760
+- Total number of images: ~36224
 - Total number of aspect buckets: 1
 - Resolution: 1.0 megapixels
 - Cropped: True

optimizer.bin

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

random_states_0.pkl

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

scheduler.bin

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

training_state-dalle3.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2f4064647ba70bb5feed84a9c3fc23f78ba95188059c70dd572b2324355817cf
-size 8370493
+oid sha256:e1696ff4dffbac7767ac8dfcfdd92c15757502966690fb753db6451e462c231e
+size 8432052

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:aa82b9ec4b918a5f369d90908b6c28169ecba097695a974663e3c4924828472b
-size 2526106
+oid sha256:2b32ea2739f91f59b6f627a2c9f5988f623469756f3c086d22be242ddb29d19c
+size 2577834

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0caa3868cfa2015e84ffe81fcd0d2323d7ad65701211cea195406d8e44b99a52
-size 2673074
+oid sha256:ea2fc9631ddd047c286926ba01b84955d6550339b466fa74fc30b7739d8affd9
+size 2726074

training_state.json

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

transformer/diffusion_pytorch_model.safetensors

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