Trained for 0 epochs and 5800 steps.

Trained with datasets ['text-embeds-sdxl', 'photocb-clip-embeds', 'celebrities', 'movieposters', 'normalnudes', 'propagandaposters', 'guys', 'pixel-art', 'signs', 'moviecollection', 'bookcovers', 'nijijourney', 'experimental', 'ethnic', 'sports', 'gay', 'architecture', 'shutterstock', 'cinemamix-1mp', 'nsfw-1024', 'anatomy', 'bg20k-1024', 'yoga', 'photo-aesthetics', 'text-1mp', 'photo-concept-bucket']
Learning rate 1e-06, batch size 8, and 4 gradient accumulation steps.
Used DDPM noise scheduler for training with v_prediction prediction type and rescaled_betas_zero_snr=True
Using 'trailing' timestep spacing.
Base model: ptx0/terminus-xl-velocity-v1
VAE: madebyollin/sdxl-vae-fp16-fix

README.md

@@ -44,7 +44,7 @@ You may reuse the base model text encoder for inference.
 ## Training settings
 
 - Training epochs: 0
-- Training steps: 5700
+- Training steps: 5800
 - Learning rate: 1e-06
 - Effective batch size: 32
   - Micro-batch size: 8
@@ -213,7 +213,7 @@ You may reuse the base model text encoder for inference.
 ### bg20k-1024
 - Repeats: 0
 - Total number of images: 89250
-- Total number of aspect buckets: 2
+- Total number of aspect buckets: 3
 - Resolution: 1.0 megapixels
 - Cropped: True
 - Crop style: random
@@ -235,7 +235,7 @@ You may reuse the base model text encoder for inference.
 - Crop style: random
 - Crop aspect: random
 ### text-1mp
-- Repeats: 5
+- Repeats: 25
 - Total number of images: 13123
 - Total number of aspect buckets: 3
 - Resolution: 1.0 megapixels

optimizer.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:38aaf03f1d0f022e0ded403a73c56f2a3b94598ba1c912ef90f9d1c9a8b5da67
+oid sha256:b2955a06ec624e837ad3705e23ce727473341d891fd0b9a8bdc2211e648e145a
 size 15406336826

random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fed20fdaa3c90c388854bcd58c825b64158299f62664a3955c6a0e08f0e36b5b
+oid sha256:946b5e8032258dd0529d94abcd17e9115d4b1baad482f372052f9f3938b62c1d
 size 14344

scheduler.bin

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

training_state-bg20k-1024.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9dc8f8727ba8b117bc5fedda7c9ca1cf457dce8fa7291c5220b9dfe427a1d153
-size 16221299
+oid sha256:66e67db2db7881eee6610fc3ba7f3b16d038d63ceb11f5776d9828c5cdc20998
+size 16310449

training_state-photo-concept-bucket.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2b2f1bb735a646e04d330efb74ca1fc1a423c57b0412e32a562e356116535bb0
-size 15482899
+oid sha256:ef9906eb25f3120400a4a2205156f8a0526f516b9eeba6ed92a1831e5d92d4c3
+size 15501702

training_state.json

@@ -1 +1 @@
-{"global_step": 5700, "epoch_step": 241, "epoch": 1, "exhausted_backends": ["guys", "signs", "nijijourney", "propagandaposters", "bookcovers", "pixel-art", "normalnudes", "celebrities", "sports", "movieposters", "moviecollection", "gay", "ethnic", "experimental", "yoga", "architecture", "cinemamix-1mp"], "repeats": {"guys": 0, "signs": 0, "nijijourney": 0, "propagandaposters": 0, "bookcovers": 0, "pixel-art": 0, "normalnudes": 0, "celebrities": 0, "sports": 0, "movieposters": 0, "moviecollection": 0, "gay": 0, "ethnic": 0, "experimental": 0, "yoga": 0, "architecture": 0, "cinemamix-1mp": 0}}
+{"global_step": 5800, "epoch_step": 160, "epoch": 1, "exhausted_backends": ["guys", "signs", "nijijourney", "propagandaposters", "bookcovers", "pixel-art", "normalnudes", "celebrities", "sports", "movieposters", "moviecollection", "gay", "ethnic", "experimental", "yoga", "architecture", "cinemamix-1mp"], "repeats": {"guys": 0, "signs": 0, "nijijourney": 0, "propagandaposters": 0, "bookcovers": 0, "pixel-art": 0, "normalnudes": 0, "celebrities": 0, "sports": 0, "movieposters": 0, "moviecollection": 0, "gay": 0, "ethnic": 0, "experimental": 0, "yoga": 0, "architecture": 0, "cinemamix-1mp": 0}}

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.27.2",
-  "_name_or_path": "/notebooks/datasets/models/checkpoint-5500",
+  "_name_or_path": "/notebooks/datasets/models/checkpoint-5700",
   "act_fn": "silu",
   "addition_embed_type": "text_time",
   "addition_embed_type_num_heads": 64,

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5f447d0cfdfba2f2ddf1ad3f1d81d840f865bf3637ad407ca6b94f79ece52050
+oid sha256:0457bc97dc56c993dfc8e256df6303a79dc9c0cd9d3a3fa482243beac513076a
 size 5135151440