Trained for 2 epochs and 76000 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

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

optimizer.bin

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

random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:68e8d3ac7922f78ce70f92151d2092407b77ac2117340a02bc8a1921421bda34
-size 16100
+oid sha256:1377b2855c166ed2a5a3d15f2a57b1b63f4f57823de8b023b297d1913febe82a
+size 16036

scheduler.bin

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

training_state-dalle3.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f7aa45d636d594442799fe40f17fa7cee86d7512202895520bd9eabecb4c401a
-size 15160772
+oid sha256:4d9f66b3a36e93ef1f019400297c06c6e45af515d57254c4dd95d6a52ea66357
+size 15223871

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:200eae3c06c244c191a62d4a1e889558d0d86f5ef0c384c71bc5bc22a48b52cc
-size 2501090
+oid sha256:3460cf25cc38cb5380179079f713997087de2a75176221cac9fa123814f27887
+size 2552394

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

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:69827830eb80cd1b821a79fbd707db0f196641061586a8755bc83848d9766105
-size 2652722
+oid sha256:1efcbffba88cb056a6f63b00cf694195e4794759cc28e44cd6322974627fb892
+size 2701906

training_state.json

@@ -1 +1 @@
-{"global_step": 75000, "epoch_step": 75002, "epoch": 3, "exhausted_backends": ["guys", "signs", "nijijourney", "bookcovers", "normalnudes", "celebrities", "gay", "sports", "moviecollection", "experimental", "ethnic", "cinemamix-1mp", "architecture", "yoga", "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, "id-75k": 0, "bg20k-1024": 0, "mj-60": 0, "sfwbooru": 0, "midjourney-v6-520k-raw": 0, "nijijourney-v6-520k-raw": 0}}
+{"global_step": 76000, "epoch_step": 76002, "epoch": 3, "exhausted_backends": ["guys", "signs", "nijijourney", "bookcovers", "normalnudes", "celebrities", "gay", "sports", "moviecollection", "experimental", "ethnic", "cinemamix-1mp", "architecture", "yoga", "nsfw-1024", "shutterstock", "photo-aesthetics", "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": 1, "text-1mp": 2, "nsfw-1024": 0, "shutterstock": 0, "id-75k": 0, "bg20k-1024": 0, "mj-60": 0, "sfwbooru": 0, "midjourney-v6-520k-raw": 0, "nijijourney-v6-520k-raw": 0}}

transformer/diffusion_pytorch_model.safetensors

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