Add converted safetensors and original pth weights
Browse files- FontDiffuser_training_phase_1_config.yaml +64 -0
- global_step_0/content_encoder.safetensors +3 -0
- global_step_0/style_encoder.safetensors +3 -0
- global_step_0/total_model.safetensors +3 -0
- global_step_0/unet.safetensors +3 -0
- global_step_1000/content_encoder.safetensors +3 -0
- global_step_1000/style_encoder.safetensors +3 -0
- global_step_1000/total_model.safetensors +3 -0
- global_step_1000/unet.safetensors +3 -0
- global_step_1500/content_encoder.safetensors +3 -0
- global_step_1500/style_encoder.safetensors +3 -0
- global_step_1500/total_model.safetensors +3 -0
- global_step_1500/unet.safetensors +3 -0
- global_step_2000/content_encoder.safetensors +3 -0
- global_step_2000/style_encoder.safetensors +3 -0
- global_step_2000/total_model.safetensors +3 -0
- global_step_2000/unet.safetensors +3 -0
- global_step_500/content_encoder.safetensors +1 -1
- global_step_500/style_encoder.safetensors +1 -1
- global_step_500/total_model.safetensors +1 -1
- global_step_500/unet.safetensors +1 -1
FontDiffuser_training_phase_1_config.yaml
ADDED
|
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
adam_beta1: 0.9
|
| 2 |
+
adam_beta2: 0.999
|
| 3 |
+
adam_epsilon: 1.0e-08
|
| 4 |
+
adam_weight_decay: 0.01
|
| 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 |
+
correcting_x0_fn: null
|
| 15 |
+
data_root: my_dataset
|
| 16 |
+
drop_prob: 0.1
|
| 17 |
+
enable_style_transform: false
|
| 18 |
+
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: 0
|
| 24 |
+
log_interval: 50
|
| 25 |
+
logging_dir: logs
|
| 26 |
+
lr_scheduler: cosine
|
| 27 |
+
lr_warmup_steps: 200
|
| 28 |
+
max_grad_norm: 1.0
|
| 29 |
+
max_train_steps: 2000
|
| 30 |
+
method: multistep
|
| 31 |
+
mixed_precision: fp16
|
| 32 |
+
mode: refinement
|
| 33 |
+
model_type: noise
|
| 34 |
+
nce_layers: 0,1,2,3
|
| 35 |
+
num_inference_steps: 20
|
| 36 |
+
num_neg: 16
|
| 37 |
+
offset_coefficient: 0.5
|
| 38 |
+
order: 2
|
| 39 |
+
output_dir: outputs/FontDiffuser
|
| 40 |
+
perceptual_coefficient: 0.05
|
| 41 |
+
phase_1_ckpt_dir: null
|
| 42 |
+
phase_2: false
|
| 43 |
+
report_to: wandb
|
| 44 |
+
resolution: 96
|
| 45 |
+
sc_coefficient: 0.01
|
| 46 |
+
scale_lr: false
|
| 47 |
+
scr_ckpt_path: null
|
| 48 |
+
scr_image_size: 96
|
| 49 |
+
seed: 123
|
| 50 |
+
skip_type: time_uniform
|
| 51 |
+
style_image_size: !!python/tuple
|
| 52 |
+
- 96
|
| 53 |
+
- 96
|
| 54 |
+
style_start_channel: 64
|
| 55 |
+
t_end: null
|
| 56 |
+
t_start: null
|
| 57 |
+
temperature: 0.07
|
| 58 |
+
train_batch_size: 16
|
| 59 |
+
unet_channels: !!python/tuple
|
| 60 |
+
- 64
|
| 61 |
+
- 128
|
| 62 |
+
- 256
|
| 63 |
+
- 512
|
| 64 |
+
val_interval: 100
|
global_step_0/content_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:46c495193ce931d5f034b409c937d6b8475f1674b5d0a31b35c592975d91202a
|
| 3 |
+
size 4756580
|
global_step_0/style_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:63085968e544836920fe7ca70b88b5face513421910710e70115d5080260c2ad
|
| 3 |
+
size 82394556
|
global_step_0/total_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5a24159e6adec280867e8f3fa62f6d92d9afb8ff70bbb4c601a8a917637a1afc
|
| 3 |
+
size 402084924
|
global_step_0/unet.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6683ab8530ebe72c753b9f4c791322c6890f581034ee46810c47c5c64ef0b122
|
| 3 |
+
size 314927748
|
global_step_1000/content_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:84ae59da8491bd7e950eae1efb8a91f82ad73b8ecd860edb9833f20f3036934f
|
| 3 |
+
size 4756580
|
global_step_1000/style_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:90d150a246490d5c21e34c128765901721de71e1ab484dfcc2667a2c6afa14bb
|
| 3 |
+
size 82394556
|
global_step_1000/total_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b0f0cfec7db400b3b6e142d1682ffe32acaca5af42b4224dbc169ead518f45c8
|
| 3 |
+
size 402084924
|
global_step_1000/unet.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:819f540cb25a50b498fa3b3054185ff3e537bc370c05ec41daba3fa8a205e54f
|
| 3 |
+
size 314927748
|
global_step_1500/content_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3601fb5bd50e58ace98f0ceded4686a72bb01de93c8d1ad2f5117ccfe30bc3f0
|
| 3 |
+
size 4756580
|
global_step_1500/style_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:91f5bd0fa4a82504d2e4c7a4cc310fea6bfd32e4c78c3f3c127f827e6a4b525d
|
| 3 |
+
size 82394556
|
global_step_1500/total_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:69dcecfed9174930a2f98c187484f5d0f05552a008a53b239bf9b62eb10c8f20
|
| 3 |
+
size 402084924
|
global_step_1500/unet.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fa7003592197f1058c7581bd102e125fab639bacf43222d809a64b0d1c905819
|
| 3 |
+
size 314927748
|
global_step_2000/content_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a5bd927428afb7f04e9109198b4d9147558eb02b7ef1bfc7ff89ee1551a0f858
|
| 3 |
+
size 4756580
|
global_step_2000/style_encoder.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e1487cdd5659b3514a01f1488f869b4cabf23a88d6a04faf6e71fc4853cfc645
|
| 3 |
+
size 82394556
|
global_step_2000/total_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b0cb119da92d54609f90fc121092478abbe9fb45542a059eb3254cc7170fdcc6
|
| 3 |
+
size 402084924
|
global_step_2000/unet.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:77a96e84c33e8050f53a8f6dc91b18b1ede4f726cb08b364052c656012e7a3c2
|
| 3 |
+
size 314927748
|
global_step_500/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:4f444844edf115917a3f0a429a03ba1e5325e6f3292fef93c8da72dfdcd10b67
|
| 3 |
size 4756580
|
global_step_500/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:9d7e714cdcbc33416765891a9d5a8f9aacf17e3160a024704a891c7661638108
|
| 3 |
size 82394556
|
global_step_500/total_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 402084924
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f47771f57c72cdbe92322f257dbc4e89ef6b9a0e0ea840030cd2b88d2e9e2edb
|
| 3 |
size 402084924
|
global_step_500/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:b8e50f6c85e674a8be6fbc41479fd0958e8ac48fe0349e6998e181865cbe53b3
|
| 3 |
size 314927748
|