Trained for 0 epochs and 20000 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 (12) 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-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.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: 0
-- Training steps: 19000
 - Learning rate: 1e-06
 - Effective batch size: 64
  - Micro-batch size: 8

 ## Training settings
 - Training epochs: 0
+- Training steps: 20000
 - 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:138d3129ba53586f00c5ef19af088217b7dfa89d3a70535cad4c52f12a5ab5a9
 size 5451415117

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

random_states_0.pkl CHANGED Viewed

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

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

scheduler.bin CHANGED Viewed

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

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

training_state-anatomy.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:428241e0d53fe5521055f1deb0be173f07a179d68adb80cdb9fb444cad100c1a
-size 9397227

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

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

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

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

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:99fa7fa654e4964349374d6a601e6413fa4cd09029d16b999c517eea5a5b4604
-size 3513018

 version https://git-lfs.github.com/spec/v1
+oid sha256:4a720a053df82130dd141a37ea87cf5be21ad454d180bed4a2d0376a6bfbe81b
+size 3603330

training_state-sfwbooru.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": ~~19000~~, "epoch_step": ~~19000~~, "epoch": 1, "exhausted_backends": ["nijijourney", "signs", "guys", "celebrities", "bookcovers", "normalnudes", "moviecollection", "sports", "gay", "experimental", "yoga", "ethnic", "architecture", "cinemamix-1mp", "photo-aesthetics", "nsfw-1024", "shutterstock", "id-75k", "text-1mp", "bg20k-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": 5, "text-1mp": 0, "nsfw-1024": 0, "shutterstock": 0, "id-75k": 0, "bg20k-1024": 0}}

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

transformer/diffusion_pytorch_model.safetensors CHANGED Viewed

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

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