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

random_states_0.pkl

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

scheduler.bin

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

training_state-dalle3.json

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

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

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

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

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

training_state.json

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

transformer/diffusion_pytorch_model.safetensors

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