PseudoTerminal X commited on
Commit
0a788ce
1 Parent(s): a2fa48e

Trained for 0 epochs and 9500 steps.

Browse files

Trained 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 CHANGED
@@ -14,7 +14,7 @@ inference: true
14
 
15
  # terminus-xl-velocity-training
16
 
17
- This is a full rank finetuned model 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,40 +44,33 @@ You may reuse the base model text encoder for inference.
44
  ## Training settings
45
 
46
  - Training epochs: 0
47
- - Training steps: 9000
48
  - Learning rate: 4e-07
49
- - Effective batch size: 16
50
- - Micro-batch size: 4
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: Not used
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: 1
71
- - Total number of images: 57744
72
- - Total number of aspect buckets: 1
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: 70616
80
- - Total number of aspect buckets: 4
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:1c5fb4a52cbf58218b3e0ca3261904c233bf6895f9f9cfffbfeabd203026c139
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:df132266404cad7e3daf43793b9b1bbccb66243137d04410e452d5b75dcc63f7
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:8eb730670117893a941498760651c7bb31af112047a438fe7509a55386824f73
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:c35d2aba7a0651ff48f997b1cdfdedd4e2209063a28dfe9e04df3b60dd2975c4
3
- size 9128699
 
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": 9000, "epoch_step": 6923, "epoch": 1, "exhausted_backends": ["celebrities", "pixel-art", "movieposters", "moviecollection", "training-test"], "repeats": {"celebrities": 0, "pixel-art": 0, "movieposters": 0, "moviecollection": 0, "training-test": 0, "dalle3": 1}}
 
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-2200",
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:81e8ca245a5180d0279bc92299a27a77095827e33c9a933f8e9ed8edb230daf7
3
  size 5135151416
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:080d173458538fecbf28799833e51197017f0ab2f6ee23291369ce1a8ecf4720
3
  size 5135151416