dzungpham commited on
Commit
559d631
·
verified ·
1 Parent(s): 5d0a97a

Add converted safetensors and original pth weights

Browse files
FontDiffuser_training_phase_2_config.yaml ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 250
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
+ experience_name: FontDiffuser_training_phase_2
18
+ gradient_accumulation_steps: 2
19
+ guidance_scale: 7.5
20
+ guidance_type: classifier-free
21
+ learning_rate: 1.0e-05
22
+ local_rank: -1
23
+ log_interval: 50
24
+ logging_dir: logs
25
+ lr_scheduler: constant
26
+ lr_warmup_steps: 100
27
+ max_grad_norm: 1.0
28
+ max_train_steps: 500
29
+ method: multistep
30
+ mixed_precision: 'no'
31
+ mode: refinement
32
+ model_type: noise
33
+ nce_layers: 0,1,2,3
34
+ num_inference_steps: 20
35
+ num_neg: 11
36
+ offset_coefficient: 0.5
37
+ order: 2
38
+ output_dir: outputs/FontDiffuser
39
+ perceptual_coefficient: 0.05
40
+ phase_1_ckpt_dir: ckpt
41
+ phase_2: true
42
+ report_to: wandb
43
+ resolution: 96
44
+ sc_coefficient: 0.05
45
+ scale_lr: false
46
+ scr_ckpt_path: ckpt/scr_210000.pth
47
+ scr_image_size: 96
48
+ seed: 123
49
+ skip_type: time_uniform
50
+ style_image_size: !!python/tuple
51
+ - 96
52
+ - 96
53
+ style_start_channel: 64
54
+ t_end: null
55
+ t_start: null
56
+ temperature: 0.07
57
+ train_batch_size: 8
58
+ unet_channels: !!python/tuple
59
+ - 64
60
+ - 128
61
+ - 256
62
+ - 512
63
+ val_interval: 100
global_step_250/content_encoder.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:41a5450edc95ae939fd8c8de7661c91718fd759f6eff96db72c14d50999ada35
3
+ size 4756580
global_step_250/style_encoder.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57d564564a3d45b0b12a7e5789e7386bf98f5283353e48ee675bd94ca7f87512
3
+ size 82394556
global_step_250/total_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:33efd840098318779a59a1c5964a19e3976596628757956df43b66729110da87
3
+ size 402084924
global_step_250/unet.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:848033aa569e5ca44bcddc8cd203f366fccdd9d75b971e14b509b493413323a6
3
+ size 314927748
global_step_500/content_encoder.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8117e2ade6b2516c520383a44ccdf2ff76de367471eb34273666110b174e4b51
3
+ size 4756580
global_step_500/style_encoder.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:72b655646b782c59c7a487313e979edf5d70e346884a591f38c76bb89ba25076
3
+ size 82394556
global_step_500/total_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0fea92847f98fbda5884aae63513b364081bac20f28282c0c88ddc47e7ac3b70
3
+ size 402084924
global_step_500/unet.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5f1073e2a73ebb81d0dcaa8e2fb206b929fc9688871529a570c329483637214
3
+ size 314927748