Trained for 4 epochs and 145000 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: 4
-- Training steps: 144000
 - Learning rate: 1e-06
 - Effective batch size: 64
  - Micro-batch size: 8

 ## Training settings
 - Training epochs: 4
+- Training steps: 145000
 - 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:e5249eecd4559d26bb0888c5bd89e2488a8b8083dc5eed8c55a0553b80c19b9a
 size 5451415117

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

random_states_0.pkl CHANGED Viewed

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

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

scheduler.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:c790cf2b6f2d50332263670578d1a3b816e5c63b48ea37599d8d83f5f9395388
 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:07a3e974cc1eb9caaee28346e432369926bacca56090cdac425ee51e3697f1bd
-size 11931022

 version https://git-lfs.github.com/spec/v1
+oid sha256:333f01b6a0258f612d45d3147ffb4918d2180575cd389cb5cf6a98ab02200ea4
+size 11998155

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

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:815ad667c482fb4e20757334eb35312e7e53b9238e82d6100685d232b58adf56
+size 2700794

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:275c64d55c0e6a3242422671598cc39d83993295a8dc6af5124e18cfc5ef0b2f
-size 2786282

 version https://git-lfs.github.com/spec/v1
+oid sha256:7eb0761f64a892ddc8bd78266142428acbe271e6272bdc0a320670777020542b
+size 2840978

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": ~~144000~~, "epoch_step": ~~144004~~, "epoch": 5, "exhausted_backends": ["signs", "bookcovers", "guys", "sports", "normalnudes", "nijijourney", "celebrities", "moviecollection", "gay", "experimental", "ethnic", "cinemamix-1mp", "yoga", "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": 3, "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": 145000, "epoch_step": 145004, "epoch": 5, "exhausted_backends": ["signs", "bookcovers", "guys", "sports", "normalnudes", "nijijourney", "celebrities", "moviecollection", "gay", "experimental", "ethnic", "cinemamix-1mp", "yoga", "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}}

transformer/diffusion_pytorch_model.safetensors CHANGED Viewed

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

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