Finish training for 12k steps for phase 1 and 10k steps for phase 2
Browse files- phase1/FontDiffuser_training_phase_1_config.yaml +6 -4
- phase1/checkpoint_step_1000/content_encoder.safetensors +3 -0
- phase1/checkpoint_step_1000/style_encoder.safetensors +3 -0
- phase1/checkpoint_step_1000/training_state.pt +3 -0
- phase1/checkpoint_step_1000/unet.safetensors +3 -0
- phase1/checkpoint_step_1500/content_encoder.safetensors +1 -1
- phase1/checkpoint_step_1500/style_encoder.safetensors +1 -1
- phase1/checkpoint_step_1500/training_state.pt +1 -1
- phase1/checkpoint_step_1500/unet.safetensors +1 -1
- phase1/checkpoint_step_2000/content_encoder.safetensors +3 -0
- phase1/checkpoint_step_2000/style_encoder.safetensors +3 -0
- phase1/checkpoint_step_2000/training_state.pt +3 -0
- phase1/checkpoint_step_2000/unet.safetensors +3 -0
- phase1/checkpoint_step_500/content_encoder.safetensors +3 -0
- phase1/checkpoint_step_500/style_encoder.safetensors +3 -0
- phase1/checkpoint_step_500/training_state.pt +3 -0
- phase1/checkpoint_step_500/unet.safetensors +3 -0
- phase1/final/content_encoder.safetensors +1 -1
- phase1/final/style_encoder.safetensors +1 -1
- phase1/final/training_state.pt +1 -1
- phase1/final/unet.safetensors +1 -1
phase1/FontDiffuser_training_phase_1_config.yaml
CHANGED
|
@@ -5,12 +5,13 @@ adam_weight_decay: 0.01
|
|
| 5 |
algorithm_type: dpmsolver++
|
| 6 |
beta_scheduler: scaled_linear
|
| 7 |
channel_attn: true
|
| 8 |
-
ckpt_interval:
|
| 9 |
content_encoder_downsample_size: 3
|
| 10 |
content_image_size: !!python/tuple
|
| 11 |
- 96
|
| 12 |
- 96
|
| 13 |
content_start_channel: 64
|
|
|
|
| 14 |
correcting_x0_fn: null
|
| 15 |
data_root: my_dataset
|
| 16 |
drop_prob: 0.1
|
|
@@ -19,6 +20,7 @@ experience_name: FontDiffuser_training_phase_1
|
|
| 19 |
gradient_accumulation_steps: 2
|
| 20 |
guidance_scale: 7.5
|
| 21 |
guidance_type: classifier-free
|
|
|
|
| 22 |
learning_rate: 0.0001
|
| 23 |
local_rank: -1
|
| 24 |
log_interval: 50
|
|
@@ -26,7 +28,7 @@ logging_dir: logs
|
|
| 26 |
lr_scheduler: cosine
|
| 27 |
lr_warmup_steps: 200
|
| 28 |
max_grad_norm: 1.0
|
| 29 |
-
max_train_steps:
|
| 30 |
method: multistep
|
| 31 |
mixed_precision: fp16
|
| 32 |
mode: refinement
|
|
@@ -39,8 +41,8 @@ offset_coefficient: 0.5
|
|
| 39 |
order: 2
|
| 40 |
output_dir: outputs/FontDiffuser/phase1
|
| 41 |
perceptual_coefficient: 0.05
|
| 42 |
-
phase_1:
|
| 43 |
-
phase_1_ckpt_dir: ckpt/
|
| 44 |
phase_2: false
|
| 45 |
report_to: wandb
|
| 46 |
resolution: 96
|
|
|
|
| 5 |
algorithm_type: dpmsolver++
|
| 6 |
beta_scheduler: scaled_linear
|
| 7 |
channel_attn: true
|
| 8 |
+
ckpt_interval: 500
|
| 9 |
content_encoder_downsample_size: 3
|
| 10 |
content_image_size: !!python/tuple
|
| 11 |
- 96
|
| 12 |
- 96
|
| 13 |
content_start_channel: 64
|
| 14 |
+
controlnet: false
|
| 15 |
correcting_x0_fn: null
|
| 16 |
data_root: my_dataset
|
| 17 |
drop_prob: 0.1
|
|
|
|
| 20 |
gradient_accumulation_steps: 2
|
| 21 |
guidance_scale: 7.5
|
| 22 |
guidance_type: classifier-free
|
| 23 |
+
instructpix2pix: false
|
| 24 |
learning_rate: 0.0001
|
| 25 |
local_rank: -1
|
| 26 |
log_interval: 50
|
|
|
|
| 28 |
lr_scheduler: cosine
|
| 29 |
lr_warmup_steps: 200
|
| 30 |
max_grad_norm: 1.0
|
| 31 |
+
max_train_steps: 2000
|
| 32 |
method: multistep
|
| 33 |
mixed_precision: fp16
|
| 34 |
mode: refinement
|
|
|
|
| 41 |
order: 2
|
| 42 |
output_dir: outputs/FontDiffuser/phase1
|
| 43 |
perceptual_coefficient: 0.05
|
| 44 |
+
phase_1: true
|
| 45 |
+
phase_1_ckpt_dir: ckpt/phase2/final
|
| 46 |
phase_2: false
|
| 47 |
report_to: wandb
|
| 48 |
resolution: 96
|
phase1/checkpoint_step_1000/content_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:eaec8a48e134ceb0c3cb2396c94dd3ef41ed837bb3fae0ab60a997f3b3dd5efd
|
| 3 |
+
size 4756580
|
phase1/checkpoint_step_1000/style_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2442e33c9d9a09a4908bb0fba19b3f1a2fcf2a39c54566a6112e4963922c2ce2
|
| 3 |
+
size 82394556
|
phase1/checkpoint_step_1000/training_state.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:33437cfba48ac32badbe8c4155e591f1155b6f5f051706fb49800ae967a0ad62
|
| 3 |
+
size 1206964193
|
phase1/checkpoint_step_1000/unet.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2a9f6488b2c510a60897aa406f88b5a41198324f1bdd92cb1fc421af35f59c3a
|
| 3 |
+
size 314927748
|
phase1/checkpoint_step_1500/content_encoder.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 4756580
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:550412b45bbabe16c827b4e757e53ed6c6f9cda93cfc7ca7e3529351fd0ca2d5
|
| 3 |
size 4756580
|
phase1/checkpoint_step_1500/style_encoder.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 82394556
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cf147e071f7e30d0e60ec58d10786c3c6aab7e827ae916b8cd9b4c2bf7b3036f
|
| 3 |
size 82394556
|
phase1/checkpoint_step_1500/training_state.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1206964193
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d5d3506cb537da83d9cf545ebb657ebb940312142540f797cbb12f2ab3a1040f
|
| 3 |
size 1206964193
|
phase1/checkpoint_step_1500/unet.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 314927748
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f3da5e0959c45ce1d372ca77beb62f787f039da2ef888b9cca8f113e762d2f5a
|
| 3 |
size 314927748
|
phase1/checkpoint_step_2000/content_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ce049b1d63e42185139684d2e0d82e92e0d5552d4f374bd4fa2206b221c765c0
|
| 3 |
+
size 4756580
|
phase1/checkpoint_step_2000/style_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7174ff24b6d04ecf2b78ed7477d36cbad18e205b879e9a4015b9cc671e832d33
|
| 3 |
+
size 82394556
|
phase1/checkpoint_step_2000/training_state.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b8b76d4d5526af0042c7bc1d478e755213c55d5c6cd3191810e7e28a1f829ead
|
| 3 |
+
size 1206964193
|
phase1/checkpoint_step_2000/unet.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9b020777f848e103300e9695204012c10c1d4b29afda5965d5adc4b92fc209fc
|
| 3 |
+
size 314927748
|
phase1/checkpoint_step_500/content_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f83259efbb9f9b6e22cf3f8087dc1853331c8fa70baa2b945ec71db8ce3a844c
|
| 3 |
+
size 4756580
|
phase1/checkpoint_step_500/style_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:706132b77b57aa80a0aded0a5ef8ddc5ea5a772a48808e658cc5b63d319d7054
|
| 3 |
+
size 82394556
|
phase1/checkpoint_step_500/training_state.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:01acf21d27acbb7fcb08abd8c4e4552310f15060265bbe08926b00da217aa6e9
|
| 3 |
+
size 1206964193
|
phase1/checkpoint_step_500/unet.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dfcf53b1817eff3eb06d68a7114f95f4a1b9b0cd55172b9414c5b7a6fe61b055
|
| 3 |
+
size 314927748
|
phase1/final/content_encoder.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 4756580
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ce049b1d63e42185139684d2e0d82e92e0d5552d4f374bd4fa2206b221c765c0
|
| 3 |
size 4756580
|
phase1/final/style_encoder.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 82394556
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7174ff24b6d04ecf2b78ed7477d36cbad18e205b879e9a4015b9cc671e832d33
|
| 3 |
size 82394556
|
phase1/final/training_state.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1206964193
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b8b76d4d5526af0042c7bc1d478e755213c55d5c6cd3191810e7e28a1f829ead
|
| 3 |
size 1206964193
|
phase1/final/unet.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 314927748
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9b020777f848e103300e9695204012c10c1d4b29afda5965d5adc4b92fc209fc
|
| 3 |
size 314927748
|