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

 ## Training settings
 - Training epochs: 1
+- Training steps: 48000
 - 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:1cd1c2c553afa3f4896641f6408d579b8292077df1eb7e26c26a9a799c073f8f
 size 5451415117

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

random_states_0.pkl CHANGED Viewed

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

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

scheduler.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:bed1d0f2c1bba78567b25787b781408c16b1a278ff645205b91b02f267bdc2a4
 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:adea542f1c1126f8d0dc5d93546ce0ab1e9d5c54a491a2e8cef3aae14acb024a
-size 12220214

 version https://git-lfs.github.com/spec/v1
+oid sha256:7472c2cf5b5c982bcf24c65b9edc27c6e163fa3b0116127fd873a2c399e9ec1d
+size 12306395

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

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:637bf6bd755cfc85b0c1126fa77a767c7012e4ae75f6775c4f18bd56389c47b2
+size 2960282

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:4d652231ef2b30f99c546844e1d5b9b8077994bbf2b34b81245030632db44147
-size 3007186

 version https://git-lfs.github.com/spec/v1
+oid sha256:1f799f880897a787f030cfbd1c100220fed2eeca69bb82f13b519ab89be6b09c
+size 3083506

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": ~~47000~~, "epoch_step": ~~47001~~, "epoch": 2, "exhausted_backends": ["guys", "signs", "nijijourney", "normalnudes", "bookcovers", "celebrities", "moviecollection", "gay", "sports", "experimental", "ethnic", "cinemamix-1mp", "yoga", "architecture", "shutterstock", "nsfw-1024", "photo-aesthetics", "id-75k", "text-1mp"], "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": 2, "text-1mp": 0, "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": 48000, "epoch_step": 48001, "epoch": 2, "exhausted_backends": ["guys", "signs", "nijijourney", "normalnudes", "bookcovers", "celebrities", "moviecollection", "gay", "sports", "experimental", "ethnic", "cinemamix-1mp", "yoga", "architecture", "shutterstock", "nsfw-1024", "photo-aesthetics", "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": 2, "text-1mp": 0, "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 CHANGED Viewed

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

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