Trained for 0 epochs and 6000 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: 5000
+- Training steps: 6000
 - Learning rate: 1e-06
 - Effective batch size: 64
   - Micro-batch size: 8
@@ -192,7 +192,7 @@ You may reuse the base model text encoder for inference.
 - Crop aspect: square
 ### shutterstock
 - Repeats: 0
-- Total number of images: ~29400
+- Total number of images: ~21056
 - 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:015ba89b6024b4f2b0fa467e77f8cfe58b3c432fba594575729b8f045a92baf1
+oid sha256:b5d1c4e191ea41e45d63c93df510ef0134d3c97205cd5dd49056a4412611f847
 size 5451415117

random_states_0.pkl

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

scheduler.bin

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

training_state-dalle3.json

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

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5e727c72c5b0bd52c3a202242ed604b62e336fda1992bb35581b07eb4705177c
-size 2476922
+oid sha256:aa82b9ec4b918a5f369d90908b6c28169ecba097695a974663e3c4924828472b
+size 2526106

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

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

training_state.json

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

transformer/diffusion_pytorch_model.safetensors

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