PseudoTerminal X commited on
Commit
4c80f66
1 Parent(s): 0300adf

Trained for 0 epochs and 5500 steps.

Browse files

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 CHANGED
@@ -44,8 +44,8 @@ You may reuse the base model text encoder for inference.
44
  ## Training settings
45
 
46
  - Training epochs: 0
47
- - Training steps: 5200
48
- - Learning rate: 4e-07
49
  - Effective batch size: 32
50
  - Micro-batch size: 8
51
  - Gradient accumulation steps: 4
@@ -188,7 +188,7 @@ You may reuse the base model text encoder for inference.
188
  - Crop aspect: random
189
  ### cinemamix-1mp
190
  - Repeats: 0
191
- - Total number of images: 9009
192
  - Total number of aspect buckets: 3
193
  - Resolution: 1.0 megapixels
194
  - Cropped: True
@@ -213,7 +213,7 @@ You may reuse the base model text encoder for inference.
213
  ### bg20k-1024
214
  - Repeats: 0
215
  - Total number of images: 89250
216
- - Total number of aspect buckets: 3
217
  - Resolution: 1.0 megapixels
218
  - Cropped: True
219
  - Crop style: random
 
44
  ## Training settings
45
 
46
  - Training epochs: 0
47
+ - Training steps: 5500
48
+ - Learning rate: 1e-06
49
  - Effective batch size: 32
50
  - Micro-batch size: 8
51
  - Gradient accumulation steps: 4
 
188
  - Crop aspect: random
189
  ### cinemamix-1mp
190
  - Repeats: 0
191
+ - Total number of images: 8992
192
  - Total number of aspect buckets: 3
193
  - Resolution: 1.0 megapixels
194
  - Cropped: True
 
213
  ### bg20k-1024
214
  - Repeats: 0
215
  - Total number of images: 89250
216
+ - Total number of aspect buckets: 2
217
  - Resolution: 1.0 megapixels
218
  - Cropped: True
219
  - Crop style: random
optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:57828af062ee0ac4f5ed2ea47e3648044321971be8592af646128a3cc7ddb95a
3
  size 15406336826
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:400ae98197b63d1e231d68434967a989d6d26f523e58ffc47f40178410d81d04
3
  size 15406336826
random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:27fcbed1792a53bf42db630992fc93ad836d09dd3626738d3a3a2cc346e0507a
3
- size 14344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e814f76188faeaa10fb633fa5d29e0ecf1a55e43150ca2ddeb469f5b6b45630
3
+ size 14408
scheduler.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1b49a773987e3f79bcd2f57c877d6ba388aa395dfb070f6869767ace9d9d61d2
3
  size 1000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1bc146e60a062a6786bb3606b7e6b2452ab72964e6a44acff076d8b2d7cd2a19
3
  size 1000
training_state-anatomy.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-architecture.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-bg20k-1024.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0726f0e161566a2083ccdd468b793fde2e6416a413fbad1fba1881b8a11a047e
3
- size 15813020
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2568ea7b7106c16029870672f1498279218d409dc03c398c2fc6039c1bbf7e1c
3
+ size 16046983
training_state-cinemamix-1mp.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-nsfw-1024.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-photo-aesthetics.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-photo-concept-bucket.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4af3949b716c14316be0a8f598e722041dd6d0f5185c1809c984cac5c53f4f02
3
- size 15398734
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:83a8cdd237f1e9980f06c80c2cc6986088b61d2fdda36a6f1cea93c66c8e9090
3
+ size 15449978
training_state-shutterstock.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-text-1mp.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state-yoga.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_state.json CHANGED
@@ -1 +1 @@
1
- {"global_step": 5200, "epoch_step": 267, "epoch": 1, "exhausted_backends": ["guys", "signs", "nijijourney", "propagandaposters", "bookcovers", "pixel-art", "normalnudes", "celebrities", "sports", "movieposters", "moviecollection", "gay", "ethnic", "experimental", "yoga", "architecture"], "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}}
 
1
+ {"global_step": 5500, "epoch_step": 166, "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 CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.27.2",
4
- "_name_or_path": "/notebooks/datasets/models/checkpoint-5000",
5
  "act_fn": "silu",
6
  "addition_embed_type": "text_time",
7
  "addition_embed_type_num_heads": 64,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.27.2",
4
+ "_name_or_path": "/notebooks/datasets/models/checkpoint-5400",
5
  "act_fn": "silu",
6
  "addition_embed_type": "text_time",
7
  "addition_embed_type_num_heads": 64,
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:afbd5a7402f9b2a3ee7fed655a29ca25e9ab04ca9c7d17b5698211dc71c24b4f
3
  size 5135151440
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f8acc6fbba00bc0971dfdb3bae5de90add1e4907c9057182ad2e54c9123195f3
3
  size 5135151440