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

 ## Training settings
 - Training epochs: 2
+- Training steps: 89000
 - 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:aab6d090d5e5c691cd0a6ef706644f8d35b9031bc97f9f76d6db99965f33d23c
 size 5451415117

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

random_states_0.pkl CHANGED Viewed

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

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

scheduler.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:873b65a14eab516419c4bad2824eab54b6eaa6e267fb65baf6d033d02d0fb84a
 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:474099d6ee3ba8c05324b54a8820d353cc5f36c59b236cd6cf5c35d120334074
-size 16222775

 version https://git-lfs.github.com/spec/v1
+oid sha256:3d3cd8b46d745c5b2a5b53b731ece31bcd3e2d0be2aed0e72fe2b8b969bcd74b
+size 16319291

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

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:6e678b4973a151cd411f00312ef2ad22fcd252dd3d388860ddeeb25461803c67
+size 3455938

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:dcc188601267d75d22d14c3739e3f04d46ff6ba299e89516cb4288d214d50922
-size 3495634

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

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": ~~88000~~, "epoch_step": ~~88002~~, "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", "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, "mj-60": 0, "sfwbooru": 0, "midjourney-v6-520k-raw": 0, "nijijourney-v6-520k-raw": 0}}

+ {"global_step": 89000, "epoch_step": 89002, "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", "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, "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:1d897984b909d2363f3a4cd82354d625922b2fe2d8cc1e25cc5faf8e108c3053
 size 1816969728

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