kmpartner commited on
Commit
49bfa89
·
verified ·
1 Parent(s): 363d554

Upload folder using huggingface_hub

Browse files
checkpoint-22800/heads.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a950b18c63957aff7240d9d3e48938beb68728b3047b0c2fcaf9fdfa515de112
3
+ size 30420614
checkpoint-22800/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fa5114cfae28b509f47e71aebfd1877b0f15230696e527a3e87a61e23d3b29f6
3
+ size 1496291698
checkpoint-22800/optimizer_1.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3df7759126362d3dffe18b068bbdccdede097ee69ba9e0aecc2b02cb9687544b
3
+ size 15431330
checkpoint-22800/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b50bf36cb0a6c629d9abaa071b42c5f834262e57ba9aaf61ad858a3f5ee8379f
3
+ size 14280
checkpoint-22800/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10a9d95d2da7005b502352a1621a25266660cb1ae66e0accdb27f3bd331d2239
3
+ size 988
checkpoint-22800/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae3265257f7f3f420c9c4d93e5d10599aebea999c8778e7cf7e05f7619a19354
3
+ size 1000
checkpoint-22800/unet/config.json ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.26.3",
4
+ "_name_or_path": "k5pcmlra2-test",
5
+ "act_fn": "silu",
6
+ "addition_embed_type": "text_time",
7
+ "addition_embed_type_num_heads": 64,
8
+ "addition_time_embed_dim": 256,
9
+ "attention_head_dim": [
10
+ 5,
11
+ 10,
12
+ 20
13
+ ],
14
+ "attention_type": "default",
15
+ "block_out_channels": [
16
+ 320,
17
+ 640,
18
+ 1280
19
+ ],
20
+ "center_input_sample": false,
21
+ "class_embed_type": null,
22
+ "class_embeddings_concat": false,
23
+ "conv_in_kernel": 3,
24
+ "conv_out_kernel": 3,
25
+ "cross_attention_dim": 2048,
26
+ "cross_attention_norm": null,
27
+ "down_block_types": [
28
+ "DownBlock2D",
29
+ "CrossAttnDownBlock2D",
30
+ "CrossAttnDownBlock2D"
31
+ ],
32
+ "downsample_padding": 1,
33
+ "dropout": 0.0,
34
+ "dual_cross_attention": false,
35
+ "encoder_hid_dim": null,
36
+ "encoder_hid_dim_type": null,
37
+ "flip_sin_to_cos": true,
38
+ "freq_shift": 0,
39
+ "in_channels": 4,
40
+ "layers_per_block": 2,
41
+ "mid_block_only_cross_attention": null,
42
+ "mid_block_scale_factor": 1,
43
+ "mid_block_type": "UNetMidBlock2D",
44
+ "norm_eps": 1e-05,
45
+ "norm_num_groups": 32,
46
+ "num_attention_heads": null,
47
+ "num_class_embeds": null,
48
+ "only_cross_attention": false,
49
+ "out_channels": 4,
50
+ "projection_class_embeddings_input_dim": 2816,
51
+ "resnet_out_scale_factor": 1.0,
52
+ "resnet_skip_time_act": false,
53
+ "resnet_time_scale_shift": "default",
54
+ "reverse_transformer_layers_per_block": null,
55
+ "sample_size": 128,
56
+ "time_cond_proj_dim": null,
57
+ "time_embedding_act_fn": null,
58
+ "time_embedding_dim": null,
59
+ "time_embedding_type": "positional",
60
+ "timestep_post_act": null,
61
+ "transformer_layers_per_block": [
62
+ 1,
63
+ 1,
64
+ 2
65
+ ],
66
+ "up_block_types": [
67
+ "CrossAttnUpBlock2D",
68
+ "CrossAttnUpBlock2D",
69
+ "UpBlock2D"
70
+ ],
71
+ "upcast_attention": null,
72
+ "use_linear_projection": true
73
+ }
checkpoint-22800/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:95475ede790389661c09d7fe4e2491b2c0e681ebbf4d46d59fbc53440bd21973
3
+ size 2980972328
logs/text2image-fine-tune/1759304798.7873716/events.out.tfevents.1759304798.9ea33e74b8da.3953.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:75da8549e8853e1ff29d3c0ec273b0ff502368a6e62088dd92469973d0f7c22c
3
+ size 2841
logs/text2image-fine-tune/1759304798.7988613/hparams.yml ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.0
5
+ adv_lr: 1.0e-05
6
+ adv_weight: 0.01
7
+ allow_tf32: true
8
+ cache_dir: null
9
+ cast_teacher_unet: false
10
+ center_crop: false
11
+ checkpointing_steps: 200
12
+ checkpoints_total_limit: 10
13
+ dataloader_num_workers: 1
14
+ enable_xformers_memory_efficient_attention: true
15
+ gradient_accumulation_steps: 10
16
+ gradient_checkpointing: true
17
+ hub_model_id: null
18
+ hub_token: null
19
+ huber_c: 0.001
20
+ learning_rate: 2.0e-06
21
+ local_rank: -1
22
+ logging_dir: logs
23
+ lora_rank: 64
24
+ loss_type: huber
25
+ lr_scheduler: constant
26
+ lr_warmup_steps: 500
27
+ max_grad_norm: 1.0
28
+ max_train_samples: 4000000
29
+ max_train_steps: 22800
30
+ mixed_precision: fp16
31
+ multiphase: 1
32
+ not_apply_cfg_solver: false
33
+ num_ddim_timesteps: 40
34
+ num_train_epochs: 114
35
+ output_dir: k5pcmlra2-test
36
+ pretrained_teacher_model: segmind/Segmind-Vega
37
+ pretrained_vae_model_name_or_path: madebyollin/sdxl-vae-fp16-fix
38
+ proportion_empty_prompts: 0
39
+ push_to_hub: false
40
+ random_flip: false
41
+ report_to: tensorboard
42
+ resolution: 512
43
+ resume_from_checkpoint: checkpoint-22600
44
+ revision: null
45
+ scale_lr: false
46
+ seed: null
47
+ teacher_revision: null
48
+ tracker_project_name: text2image-fine-tune
49
+ train_batch_size: 4
50
+ train_shards_path_or_url: null
51
+ use_8bit_adam: true
52
+ use_fix_crop_and_size: false
53
+ validation_steps: 500000
54
+ w_max: 7.0
55
+ w_min: 6.0
logs/text2image-fine-tune/events.out.tfevents.1759304798.9ea33e74b8da.3953.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:556f5a7fe69ddecdf10f55429bd2bf45b099675983bbd5f783202668331533dd
3
+ size 22388