Trained for 0 epochs and 6000 steps.

Trained with datasets ['text-embeds-pixart-filter', 'photo-concept-bucket', 'moviecollection', 'experimental', 'ethnic', 'sports', 'architecture', 'shutterstock', 'cinemamix-1mp', 'nsfw-1024', 'anatomy', 'bg20k-1024', 'yoga', 'photo-aesthetics', 'text-1mp', 'movieposters', 'normalnudes', 'pixel-art', 'signs', 'midjourney-v6-520k-raw', 'sfwbooru', 'nijijourney-v6-520k-raw', 'dalle3']
Learning rate 1e-06, batch size 24, 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-large
VAE: madebyollin/sdxl-vae-fp16-fix

README.md

@@ -77,7 +77,7 @@ You may reuse the base model text encoder for inference.
 ## Training settings
 
 - Training epochs: 0
-- Training steps: 5500
+- Training steps: 6000
 - Learning rate: 1e-06
 - Effective batch size: 192
   - Micro-batch size: 24
@@ -166,8 +166,8 @@ You may reuse the base model text encoder for inference.
 - Crop aspect: random
 ### anatomy
 - Repeats: 5
-- Total number of images: ~21056
-- Total number of aspect buckets: 1
+- Total number of images: ~15168
+- Total number of aspect buckets: 3
 - Resolution: 1.0 megapixels
 - Cropped: True
 - Crop style: random
@@ -239,7 +239,7 @@ You may reuse the base model text encoder for inference.
 ### midjourney-v6-520k-raw
 - Repeats: 0
 - Total number of images: ~671104
-- Total number of aspect buckets: 15
+- Total number of aspect buckets: 10
 - Resolution: 1.0 megapixels
 - Cropped: False
 - Crop style: None
@@ -247,7 +247,7 @@ You may reuse the base model text encoder for inference.
 ### sfwbooru
 - Repeats: 0
 - Total number of images: ~423552
-- Total number of aspect buckets: 35
+- Total number of aspect buckets: 25
 - Resolution: 1.0 megapixels
 - Cropped: False
 - Crop style: None
@@ -255,7 +255,7 @@ You may reuse the base model text encoder for inference.
 ### nijijourney-v6-520k-raw
 - Repeats: 0
 - Total number of images: ~670976
-- Total number of aspect buckets: 11
+- Total number of aspect buckets: 9
 - Resolution: 1.0 megapixels
 - Cropped: False
 - Crop style: None

optimizer.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1e9674252ce4ece4f58d923749b69f9c928be43818c0ca98efb6f3ec006138a7
+oid sha256:628f076f19a3e7d4a4802a5550c04db94e8e406729a29176d4e340749e82039f
 size 5451415117

random_states_0.pkl

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

scheduler.bin

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

training_state-photo-concept-bucket.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:737efa0eb772e42d8c1dc7d6f75cf690bff607c33cec78f4300f868140a780b3
-size 6166382
+oid sha256:cfc8fd93b0098850c4aeb5709e40838c5588c8b2e2bf75ccfc0e2552a2af1c01
+size 6317385

training_state.json

@@ -1 +1 @@
-{"global_step": 5500, "epoch_step": 1, "epoch": 1, "exhausted_backends": ["signs", "nijijourney", "pixel-art", "propagandaposters", "sports", "gay", "experimental", "yoga", "ethnic", "cinemamix-1mp", "architecture", "mj-60", "normalnudes", "bookcovers", "celebrities", "movieposters", "shutterstock", "nsfw-1024", "photo-aesthetics", "moviecollection", "bg20k-1024"], "repeats": {"bookcovers": 0, "signs": 0, "normalnudes": 0, "nijijourney": 0, "movieposters": 0, "celebrities": 0, "pixel-art": 0, "propagandaposters": 0, "sports": 0, "moviecollection": 0, "gay": 0, "experimental": 0, "yoga": 0, "ethnic": 0, "cinemamix-1mp": 0, "architecture": 0, "mj-60": 0, "text-1mp": 12, "shutterstock": 0, "nsfw-1024": 0, "photo-aesthetics": 0, "anatomy": 4, "bg20k-1024": 0}}
+{"global_step": 6000, "epoch_step": 1, "epoch": 1, "exhausted_backends": ["signs", "nijijourney", "pixel-art", "propagandaposters", "sports", "gay", "experimental", "yoga", "ethnic", "cinemamix-1mp", "architecture", "mj-60", "normalnudes", "bookcovers", "celebrities", "movieposters", "shutterstock", "nsfw-1024", "photo-aesthetics", "moviecollection", "bg20k-1024"], "repeats": {"bookcovers": 0, "signs": 0, "normalnudes": 0, "nijijourney": 0, "movieposters": 0, "celebrities": 0, "pixel-art": 0, "propagandaposters": 0, "sports": 0, "moviecollection": 0, "gay": 0, "experimental": 0, "yoga": 0, "ethnic": 0, "cinemamix-1mp": 0, "architecture": 0, "mj-60": 0, "text-1mp": 13, "shutterstock": 0, "nsfw-1024": 0, "photo-aesthetics": 0, "anatomy": 5, "bg20k-1024": 0}}

transformer/diffusion_pytorch_model.safetensors

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