Trained for 6 epochs and 212000 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 +1 -1
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: 6
-- Training steps: 211000
 - Learning rate: 1e-06
 - Effective batch size: 64
   - Micro-batch size: 8

 ## Training settings
 - Training epochs: 6
+- Training steps: 212000
 - 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:2ce33ed763aacb82a4db0f2ac98383c2dc6b223877543553b0fae6106f105e63
 size 5451415117

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

random_states_0.pkl CHANGED Viewed

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

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

scheduler.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:64b97851d329128b0dc2faa4fd27e5f9472a0ff607b2b7294d9e67b2e60a75a7
 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:c8d462f246bceb7cc7ca5276f310d1c152f3e4c93c5104075bff0acaf7f98a4e
-size 8615664

 version https://git-lfs.github.com/spec/v1
+oid sha256:02c11392701aaf1cf8f5abaeea592d132e8b96f99d2bc92b046d6525f054151a
+size 8694965

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

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:748e72023c487d976cc858031028f02890d0aaf719c239f85236c4dbe0388c2d
+size 2767786

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:9847007782a9ac2ffd3629e5eb3fe14c9b5acd326d55c53c8784bce2c0a22c94
-size 2812994

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

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": ~~211000~~, "epoch_step": ~~211006~~, "epoch": 7, "exhausted_backends": ["nijijourney", "guys", "signs", "normalnudes", "bookcovers", "celebrities", "sports", "moviecollection", "gay", "yoga", "experimental", "ethnic", "cinemamix-1mp", "architecture", "shutterstock", "nsfw-1024", "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": 4, "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, "dalle3": 0}}

+ {"global_step": 212000, "epoch_step": 212006, "epoch": 7, "exhausted_backends": ["nijijourney", "guys", "signs", "normalnudes", "bookcovers", "celebrities", "sports", "moviecollection", "gay", "yoga", "experimental", "ethnic", "cinemamix-1mp", "architecture", "shutterstock", "nsfw-1024", "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": 5, "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, "dalle3": 0}}

transformer/diffusion_pytorch_model.safetensors CHANGED Viewed

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

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