PseudoTerminal X
commited on
Commit
•
4c80f66
1
Parent(s):
0300adf
Trained for 0 epochs and 5500 steps.
Browse filesTrained 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 +4 -4
- optimizer.bin +1 -1
- random_states_0.pkl +2 -2
- scheduler.bin +1 -1
- training_state-anatomy.json +0 -0
- training_state-architecture.json +0 -0
- training_state-bg20k-1024.json +2 -2
- training_state-cinemamix-1mp.json +0 -0
- training_state-nsfw-1024.json +0 -0
- training_state-photo-aesthetics.json +0 -0
- training_state-photo-concept-bucket.json +2 -2
- training_state-shutterstock.json +0 -0
- training_state-text-1mp.json +0 -0
- training_state-yoga.json +0 -0
- training_state.json +1 -1
- unet/config.json +1 -1
- unet/diffusion_pytorch_model.safetensors +1 -1
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:
|
48 |
-
- Learning rate:
|
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:
|
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:
|
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:
|
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:
|
3 |
-
size
|
|
|
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:
|
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:
|
3 |
-
size
|
|
|
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:
|
3 |
-
size
|
|
|
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":
|
|
|
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-
|
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:
|
3 |
size 5135151440
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f8acc6fbba00bc0971dfdb3bae5de90add1e4907c9057182ad2e54c9123195f3
|
3 |
size 5135151440
|