PseudoTerminal X
commited on
Commit
•
0a788ce
1
Parent(s):
a2fa48e
Trained for 0 epochs and 9500 steps.
Browse filesTrained with datasets ['text-embeds-sdxl-nofilter', 'dalle3', 'photo-concept-bucket']
Learning rate 4e-07, batch size 16, 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-v2
VAE: madebyollin/sdxl-vae-fp16-fix
- README.md +11 -18
- optimizer.bin +1 -1
- random_states_0.pkl +1 -1
- scheduler.bin +1 -1
- training_state-dalle3.json +0 -0
- training_state-photo-concept-bucket.json +2 -2
- training_state.json +1 -1
- unet/config.json +1 -1
- unet/diffusion_pytorch_model.safetensors +1 -1
README.md
CHANGED
@@ -14,7 +14,7 @@ inference: true
|
|
14 |
|
15 |
# terminus-xl-velocity-training
|
16 |
|
17 |
-
This is a full rank
|
18 |
|
19 |
The main validation prompt used during training was:
|
20 |
|
@@ -44,40 +44,33 @@ You may reuse the base model text encoder for inference.
|
|
44 |
## Training settings
|
45 |
|
46 |
- Training epochs: 0
|
47 |
-
- Training steps:
|
48 |
- Learning rate: 4e-07
|
49 |
-
- Effective batch size:
|
50 |
-
- Micro-batch size:
|
51 |
- Gradient accumulation steps: 4
|
|
|
52 |
- Prediction type: v_prediction
|
53 |
- Rescaled betas zero SNR: True
|
54 |
- Optimizer: AdamW, stochastic bf16
|
55 |
- Precision: Pure BF16
|
56 |
-
- Xformers:
|
57 |
|
58 |
|
59 |
## Datasets
|
60 |
|
61 |
-
### training-test
|
62 |
-
- Repeats: 1
|
63 |
-
- Total number of images: 192
|
64 |
-
- Total number of aspect buckets: 17
|
65 |
-
- Resolution: 1.0 megapixels
|
66 |
-
- Cropped: True
|
67 |
-
- Crop style: corner
|
68 |
-
- Crop aspect: preserve
|
69 |
### dalle3
|
70 |
-
- Repeats:
|
71 |
-
- Total number of images:
|
72 |
-
- Total number of aspect buckets:
|
73 |
- Resolution: 1.0 megapixels
|
74 |
- Cropped: False
|
75 |
- Crop style: None
|
76 |
- Crop aspect: None
|
77 |
### photo-concept-bucket
|
78 |
- Repeats: 0
|
79 |
-
- Total number of images:
|
80 |
-
- Total number of aspect buckets:
|
81 |
- Resolution: 1.0 megapixels
|
82 |
- Cropped: True
|
83 |
- Crop style: random
|
|
|
14 |
|
15 |
# terminus-xl-velocity-training
|
16 |
|
17 |
+
This is a full rank finetune derived from [ptx0/terminus-xl-velocity-v2](https://huggingface.co/ptx0/terminus-xl-velocity-v2).
|
18 |
|
19 |
The main validation prompt used during training was:
|
20 |
|
|
|
44 |
## Training settings
|
45 |
|
46 |
- Training epochs: 0
|
47 |
+
- Training steps: 9500
|
48 |
- Learning rate: 4e-07
|
49 |
+
- Effective batch size: 512
|
50 |
+
- Micro-batch size: 16
|
51 |
- Gradient accumulation steps: 4
|
52 |
+
- Number of GPUs: 8
|
53 |
- Prediction type: v_prediction
|
54 |
- Rescaled betas zero SNR: True
|
55 |
- Optimizer: AdamW, stochastic bf16
|
56 |
- Precision: Pure BF16
|
57 |
+
- Xformers: Enabled
|
58 |
|
59 |
|
60 |
## Datasets
|
61 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
62 |
### dalle3
|
63 |
+
- Repeats: 0
|
64 |
+
- Total number of images: ~461848
|
65 |
+
- Total number of aspect buckets: 27
|
66 |
- Resolution: 1.0 megapixels
|
67 |
- Cropped: False
|
68 |
- Crop style: None
|
69 |
- Crop aspect: None
|
70 |
### photo-concept-bucket
|
71 |
- Repeats: 0
|
72 |
+
- Total number of images: ~557568
|
73 |
+
- Total number of aspect buckets: 5
|
74 |
- Resolution: 1.0 megapixels
|
75 |
- Cropped: True
|
76 |
- 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:d1f1ddd05d4c41863dabfca0c5e90e90c525836a259cfccb431643878329641f
|
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 16036
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:904dd262888fa3761ef496229f2619ad05b67a5fa23e3cf2cf78409455186bfe
|
3 |
size 16036
|
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:d3abc2992b4caa0f0747a6977a22983beb24275ae52a4edb2e93fda245503a0d
|
3 |
size 1000
|
training_state-dalle3.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:dfad78eb826fdd11841cad1087efdc6fc1869be251a3862ae05f4f0fe88448cf
|
3 |
+
size 4368036
|
training_state.json
CHANGED
@@ -1 +1 @@
|
|
1 |
-
{"global_step":
|
|
|
1 |
+
{"global_step": 9500, "epoch_step": 2231, "epoch": 1, "exhausted_backends": ["celebrities", "pixel-art", "movieposters", "moviecollection", "training-test", "photo-concept-bucket"], "repeats": {"celebrities": 0, "pixel-art": 0, "movieposters": 0, "moviecollection": 0, "training-test": 0, "dalle3": 1, "photo-concept-bucket": 0}}
|
unet/config.json
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
{
|
2 |
"_class_name": "UNet2DConditionModel",
|
3 |
"_diffusers_version": "0.29.0.dev0",
|
4 |
-
"_name_or_path": "/home/user/training/lite-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.29.0.dev0",
|
4 |
+
"_name_or_path": "/home/user/training/lite-models/checkpoint-9000",
|
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 5135151416
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:080d173458538fecbf28799833e51197017f0ab2f6ee23291369ce1a8ecf4720
|
3 |
size 5135151416
|