mmkjj commited on
Commit
d769409
·
verified ·
1 Parent(s): 99b61b3

End of training

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. checkpoint-2000/optimizer.bin +3 -0
  2. checkpoint-2000/random_states_0.pkl +3 -0
  3. checkpoint-2000/scaler.pt +3 -0
  4. checkpoint-2000/scheduler.bin +3 -0
  5. checkpoint-2000/unet/config.json +68 -0
  6. checkpoint-2000/unet/diffusion_pytorch_model.safetensors +3 -0
  7. feature_extractor/preprocessor_config.json +27 -0
  8. logs/instruct-pix2pix-xl/1739017328.9015243/events.out.tfevents.1739017328.spcc-a40g02.1673616.1 +3 -0
  9. logs/instruct-pix2pix-xl/1739017328.908869/hparams.yml +55 -0
  10. logs/instruct-pix2pix-xl/1739017900.7121582/events.out.tfevents.1739017900.spcc-a40g02.1677303.1 +3 -0
  11. logs/instruct-pix2pix-xl/1739017900.7204247/hparams.yml +55 -0
  12. logs/instruct-pix2pix-xl/1739024788.2772908/events.out.tfevents.1739024788.spcc-a40g14.445920.1 +3 -0
  13. logs/instruct-pix2pix-xl/1739024788.2843132/hparams.yml +55 -0
  14. logs/instruct-pix2pix-xl/events.out.tfevents.1739017328.spcc-a40g02.1673616.0 +3 -0
  15. logs/instruct-pix2pix-xl/events.out.tfevents.1739017900.spcc-a40g02.1677303.0 +3 -0
  16. logs/instruct-pix2pix-xl/events.out.tfevents.1739024788.spcc-a40g14.445920.0 +3 -0
  17. logs/instruct-pix2pix/1739018978.779651/events.out.tfevents.1739018978.spcc-a40g19.1239308.1 +3 -0
  18. logs/instruct-pix2pix/1739018978.7848387/hparams.yml +51 -0
  19. logs/instruct-pix2pix/1739025279.2984846/events.out.tfevents.1739025279.spcc-a40g14.448688.1 +3 -0
  20. logs/instruct-pix2pix/1739025279.304702/hparams.yml +51 -0
  21. logs/instruct-pix2pix/1739029644.28327/events.out.tfevents.1739029644.spcc-a40g14.472330.1 +3 -0
  22. logs/instruct-pix2pix/1739029644.2910244/hparams.yml +51 -0
  23. logs/instruct-pix2pix/1739029905.340061/events.out.tfevents.1739029905.spcc-a40g14.473747.1 +3 -0
  24. logs/instruct-pix2pix/1739029905.3451133/hparams.yml +51 -0
  25. logs/instruct-pix2pix/1739030361.8407564/events.out.tfevents.1739030361.spcc-a40g14.476260.1 +3 -0
  26. logs/instruct-pix2pix/1739030361.8500128/hparams.yml +51 -0
  27. logs/instruct-pix2pix/1739030438.2365324/events.out.tfevents.1739030438.spcc-a40g14.476772.1 +3 -0
  28. logs/instruct-pix2pix/1739030438.2423885/hparams.yml +51 -0
  29. logs/instruct-pix2pix/1739030577.7740219/events.out.tfevents.1739030577.spcc-a40g14.477553.1 +3 -0
  30. logs/instruct-pix2pix/1739030577.7788777/hparams.yml +51 -0
  31. logs/instruct-pix2pix/1739030713.9233325/events.out.tfevents.1739030713.spcc-a40g14.478386.1 +3 -0
  32. logs/instruct-pix2pix/1739030713.9285104/hparams.yml +51 -0
  33. logs/instruct-pix2pix/1739031020.4435735/events.out.tfevents.1739031020.spcc-a40g14.479886.1 +3 -0
  34. logs/instruct-pix2pix/1739031020.4503279/hparams.yml +51 -0
  35. logs/instruct-pix2pix/events.out.tfevents.1739018978.spcc-a40g19.1239308.0 +3 -0
  36. logs/instruct-pix2pix/events.out.tfevents.1739025279.spcc-a40g14.448688.0 +3 -0
  37. logs/instruct-pix2pix/events.out.tfevents.1739029644.spcc-a40g14.472330.0 +3 -0
  38. logs/instruct-pix2pix/events.out.tfevents.1739029905.spcc-a40g14.473747.0 +3 -0
  39. logs/instruct-pix2pix/events.out.tfevents.1739030361.spcc-a40g14.476260.0 +3 -0
  40. logs/instruct-pix2pix/events.out.tfevents.1739030438.spcc-a40g14.476772.0 +3 -0
  41. logs/instruct-pix2pix/events.out.tfevents.1739030577.spcc-a40g14.477553.0 +3 -0
  42. logs/instruct-pix2pix/events.out.tfevents.1739030713.spcc-a40g14.478386.0 +3 -0
  43. logs/instruct-pix2pix/events.out.tfevents.1739031020.spcc-a40g14.479886.0 +3 -0
  44. model_index.json +38 -0
  45. safety_checker/config.json +29 -0
  46. safety_checker/model.safetensors +3 -0
  47. scheduler/scheduler_config.json +15 -0
  48. text_encoder/config.json +25 -0
  49. text_encoder/model.safetensors +3 -0
  50. tokenizer/merges.txt +0 -0
checkpoint-2000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:653ca0da7703597e7a8fe2625b5e000c3d5e7621b7e42c4f8d870f78360fc919
3
+ size 6876842324
checkpoint-2000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8fd0ab3e9ef9727e0aabeb812322305193e784c835dc257aba348affa6fe3cec
3
+ size 14408
checkpoint-2000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c50a9cebe5d66d453d25b140738bff479749ac03e0a43597d8776bc22f6ed0c
3
+ size 988
checkpoint-2000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a65898fd80ad1f5b3504355ba22c7c0f5854b105b47e4ad00c9a7a016fda2229
3
+ size 1000
checkpoint-2000/unet/config.json ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.33.0.dev0",
4
+ "_name_or_path": "stable-diffusion-v1-5/stable-diffusion-v1-5",
5
+ "act_fn": "silu",
6
+ "addition_embed_type": null,
7
+ "addition_embed_type_num_heads": 64,
8
+ "addition_time_embed_dim": null,
9
+ "attention_head_dim": 8,
10
+ "attention_type": "default",
11
+ "block_out_channels": [
12
+ 320,
13
+ 640,
14
+ 1280,
15
+ 1280
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "class_embeddings_concat": false,
20
+ "conv_in_kernel": 3,
21
+ "conv_out_kernel": 3,
22
+ "cross_attention_dim": 768,
23
+ "cross_attention_norm": null,
24
+ "down_block_types": [
25
+ "CrossAttnDownBlock2D",
26
+ "CrossAttnDownBlock2D",
27
+ "CrossAttnDownBlock2D",
28
+ "DownBlock2D"
29
+ ],
30
+ "downsample_padding": 1,
31
+ "dropout": 0.0,
32
+ "dual_cross_attention": false,
33
+ "encoder_hid_dim": null,
34
+ "encoder_hid_dim_type": null,
35
+ "flip_sin_to_cos": true,
36
+ "freq_shift": 0,
37
+ "in_channels": 8,
38
+ "layers_per_block": 2,
39
+ "mid_block_only_cross_attention": null,
40
+ "mid_block_scale_factor": 1,
41
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
42
+ "norm_eps": 1e-05,
43
+ "norm_num_groups": 32,
44
+ "num_attention_heads": null,
45
+ "num_class_embeds": null,
46
+ "only_cross_attention": false,
47
+ "out_channels": 4,
48
+ "projection_class_embeddings_input_dim": null,
49
+ "resnet_out_scale_factor": 1.0,
50
+ "resnet_skip_time_act": false,
51
+ "resnet_time_scale_shift": "default",
52
+ "reverse_transformer_layers_per_block": null,
53
+ "sample_size": 64,
54
+ "time_cond_proj_dim": null,
55
+ "time_embedding_act_fn": null,
56
+ "time_embedding_dim": null,
57
+ "time_embedding_type": "positional",
58
+ "timestep_post_act": null,
59
+ "transformer_layers_per_block": 1,
60
+ "up_block_types": [
61
+ "UpBlock2D",
62
+ "CrossAttnUpBlock2D",
63
+ "CrossAttnUpBlock2D",
64
+ "CrossAttnUpBlock2D"
65
+ ],
66
+ "upcast_attention": false,
67
+ "use_linear_projection": false
68
+ }
checkpoint-2000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89095b273ee97b7d221ba2af430054e3a006f682951998a765b9e58be5f15843
3
+ size 3438213624
feature_extractor/preprocessor_config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "crop_size": {
3
+ "height": 224,
4
+ "width": 224
5
+ },
6
+ "do_center_crop": true,
7
+ "do_convert_rgb": true,
8
+ "do_normalize": true,
9
+ "do_rescale": true,
10
+ "do_resize": true,
11
+ "image_mean": [
12
+ 0.48145466,
13
+ 0.4578275,
14
+ 0.40821073
15
+ ],
16
+ "image_processor_type": "CLIPImageProcessor",
17
+ "image_std": [
18
+ 0.26862954,
19
+ 0.26130258,
20
+ 0.27577711
21
+ ],
22
+ "resample": 3,
23
+ "rescale_factor": 0.00392156862745098,
24
+ "size": {
25
+ "shortest_edge": 224
26
+ }
27
+ }
logs/instruct-pix2pix-xl/1739017328.9015243/events.out.tfevents.1739017328.spcc-a40g02.1673616.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f421ea09d36030d68066fb376f9ac38a5dc42b1456e64021834646d2057d1c99
3
+ size 2567
logs/instruct-pix2pix-xl/1739017328.908869/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.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 5000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ crops_coords_top_left_h: 0
12
+ crops_coords_top_left_w: 0
13
+ dataloader_num_workers: 0
14
+ dataset_config_name: null
15
+ dataset_name: fusing/instructpix2pix-1000-samples
16
+ edit_prompt_column: edit_prompt
17
+ edited_image_column: edited_image
18
+ enable_xformers_memory_efficient_attention: false
19
+ gradient_accumulation_steps: 4
20
+ gradient_checkpointing: true
21
+ hub_model_id: null
22
+ hub_token: null
23
+ learning_rate: 5.0e-05
24
+ local_rank: -1
25
+ logging_dir: logs
26
+ lr_scheduler: constant
27
+ lr_warmup_steps: 0
28
+ max_grad_norm: 1.0
29
+ max_train_samples: null
30
+ max_train_steps: 15000
31
+ mixed_precision: null
32
+ non_ema_revision: null
33
+ num_train_epochs: 239
34
+ num_validation_images: 4
35
+ original_image_column: input_image
36
+ output_dir: instruct-pix2pix-model
37
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
38
+ pretrained_vae_model_name_or_path: null
39
+ push_to_hub: true
40
+ random_flip: true
41
+ report_to: tensorboard
42
+ resolution: 256
43
+ resume_from_checkpoint: null
44
+ revision: null
45
+ scale_lr: false
46
+ seed: 42
47
+ train_batch_size: 4
48
+ train_data_dir: null
49
+ use_8bit_adam: false
50
+ use_ema: false
51
+ vae_precision: fp32
52
+ val_image_url_or_path: null
53
+ validation_prompt: null
54
+ validation_steps: 100
55
+ variant: null
logs/instruct-pix2pix-xl/1739017900.7121582/events.out.tfevents.1739017900.spcc-a40g02.1677303.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc9caa8704728c7ec23d670ac55ebbf41bbc1125eda86203a2cee27f4d73ceba
3
+ size 2567
logs/instruct-pix2pix-xl/1739017900.7204247/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.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 5000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ crops_coords_top_left_h: 0
12
+ crops_coords_top_left_w: 0
13
+ dataloader_num_workers: 0
14
+ dataset_config_name: null
15
+ dataset_name: fusing/instructpix2pix-1000-samples
16
+ edit_prompt_column: edit_prompt
17
+ edited_image_column: edited_image
18
+ enable_xformers_memory_efficient_attention: false
19
+ gradient_accumulation_steps: 4
20
+ gradient_checkpointing: true
21
+ hub_model_id: null
22
+ hub_token: null
23
+ learning_rate: 5.0e-05
24
+ local_rank: -1
25
+ logging_dir: logs
26
+ lr_scheduler: constant
27
+ lr_warmup_steps: 0
28
+ max_grad_norm: 1.0
29
+ max_train_samples: null
30
+ max_train_steps: 15000
31
+ mixed_precision: null
32
+ non_ema_revision: null
33
+ num_train_epochs: 239
34
+ num_validation_images: 4
35
+ original_image_column: input_image
36
+ output_dir: instruct-pix2pix-model
37
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
38
+ pretrained_vae_model_name_or_path: null
39
+ push_to_hub: true
40
+ random_flip: true
41
+ report_to: tensorboard
42
+ resolution: 256
43
+ resume_from_checkpoint: null
44
+ revision: null
45
+ scale_lr: false
46
+ seed: 42
47
+ train_batch_size: 4
48
+ train_data_dir: null
49
+ use_8bit_adam: false
50
+ use_ema: false
51
+ vae_precision: fp32
52
+ val_image_url_or_path: null
53
+ validation_prompt: null
54
+ validation_steps: 100
55
+ variant: null
logs/instruct-pix2pix-xl/1739024788.2772908/events.out.tfevents.1739024788.spcc-a40g14.445920.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e618064a2afb25eb040379c7463ae72ba0d9671c4e4374878c58aae712d580a
3
+ size 2567
logs/instruct-pix2pix-xl/1739024788.2843132/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.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 5000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ crops_coords_top_left_h: 0
12
+ crops_coords_top_left_w: 0
13
+ dataloader_num_workers: 0
14
+ dataset_config_name: null
15
+ dataset_name: fusing/instructpix2pix-1000-samples
16
+ edit_prompt_column: edit_prompt
17
+ edited_image_column: edited_image
18
+ enable_xformers_memory_efficient_attention: false
19
+ gradient_accumulation_steps: 4
20
+ gradient_checkpointing: true
21
+ hub_model_id: null
22
+ hub_token: null
23
+ learning_rate: 5.0e-05
24
+ local_rank: -1
25
+ logging_dir: logs
26
+ lr_scheduler: constant
27
+ lr_warmup_steps: 0
28
+ max_grad_norm: 1.0
29
+ max_train_samples: null
30
+ max_train_steps: 15000
31
+ mixed_precision: null
32
+ non_ema_revision: null
33
+ num_train_epochs: 239
34
+ num_validation_images: 4
35
+ original_image_column: input_image
36
+ output_dir: instruct-pix2pix-model
37
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
38
+ pretrained_vae_model_name_or_path: null
39
+ push_to_hub: true
40
+ random_flip: true
41
+ report_to: tensorboard
42
+ resolution: 256
43
+ resume_from_checkpoint: null
44
+ revision: null
45
+ scale_lr: false
46
+ seed: 42
47
+ train_batch_size: 4
48
+ train_data_dir: null
49
+ use_8bit_adam: false
50
+ use_ema: false
51
+ vae_precision: fp32
52
+ val_image_url_or_path: null
53
+ validation_prompt: null
54
+ validation_steps: 100
55
+ variant: null
logs/instruct-pix2pix-xl/events.out.tfevents.1739017328.spcc-a40g02.1673616.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:442edece3d668ec32b25e01c3dd316193e81890966e036495414602af1a97871
3
+ size 88
logs/instruct-pix2pix-xl/events.out.tfevents.1739017900.spcc-a40g02.1677303.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:387b386dde56fef0ac2e02f07547413c8f1ae71555051239be76bf21fe65f19b
3
+ size 88
logs/instruct-pix2pix-xl/events.out.tfevents.1739024788.spcc-a40g14.445920.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ccc8cd16a6f280ae3911d593f62175bd255900b6763008c47210e96ff15b06d4
3
+ size 88
logs/instruct-pix2pix/1739018978.779651/events.out.tfevents.1739018978.spcc-a40g19.1239308.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:49b6730993cc232fffd6e7e4ce5cb487203586e3d49bb3f1e4c8c598672e4911
3
+ size 2445
logs/instruct-pix2pix/1739018978.7848387/hparams.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 5000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: sayakpaul/instructpix2pix-1000-samples
14
+ edit_prompt_column: edit_prompt
15
+ edited_image_column: edited_image
16
+ enable_xformers_memory_efficient_attention: false
17
+ gradient_accumulation_steps: 4
18
+ gradient_checkpointing: true
19
+ hub_model_id: null
20
+ hub_token: null
21
+ learning_rate: 5.0e-05
22
+ local_rank: 0
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 0
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 15000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 469
32
+ num_validation_images: 4
33
+ original_image_column: input_image
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: stable-diffusion-v1-5/stable-diffusion-v1-5
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 4
45
+ train_data_dir: null
46
+ use_8bit_adam: false
47
+ use_ema: true
48
+ val_image_url: null
49
+ validation_epochs: 1
50
+ validation_prompt: null
51
+ variant: null
logs/instruct-pix2pix/1739025279.2984846/events.out.tfevents.1739025279.spcc-a40g14.448688.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a9effbc527dc0148cdbca64ea37bbea7b98c5ae55ccc2880913cc97ea594a8d
3
+ size 2442
logs/instruct-pix2pix/1739025279.304702/hparams.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 5000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: fusing/instructpix2pix-1000-samples
14
+ edit_prompt_column: edit_prompt
15
+ edited_image_column: edited_image
16
+ enable_xformers_memory_efficient_attention: false
17
+ gradient_accumulation_steps: 4
18
+ gradient_checkpointing: true
19
+ hub_model_id: null
20
+ hub_token: null
21
+ learning_rate: 5.0e-05
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 0
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 15000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 239
32
+ num_validation_images: 4
33
+ original_image_column: input_image
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: stable-diffusion-v1-5/stable-diffusion-v1-5
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 256
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 4
45
+ train_data_dir: null
46
+ use_8bit_adam: false
47
+ use_ema: false
48
+ val_image_url: null
49
+ validation_epochs: 1
50
+ validation_prompt: null
51
+ variant: null
logs/instruct-pix2pix/1739029644.28327/events.out.tfevents.1739029644.spcc-a40g14.472330.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f40e83ed68304ca0bfcc9f900d6105197a5a060409bd312369fadc91bf615b0
3
+ size 2448
logs/instruct-pix2pix/1739029644.2910244/hparams.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 5000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: null
14
+ edit_prompt_column: instruction
15
+ edited_image_column: edited_image
16
+ enable_xformers_memory_efficient_attention: false
17
+ gradient_accumulation_steps: 4
18
+ gradient_checkpointing: true
19
+ hub_model_id: null
20
+ hub_token: null
21
+ learning_rate: 5.0e-05
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 0
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 15000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 334
32
+ num_validation_images: 4
33
+ original_image_column: image_name
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: stable-diffusion-v1-5/stable-diffusion-v1-5
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 256
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 4
45
+ train_data_dir: ./self_barefoot_dataset/new_structure/
46
+ use_8bit_adam: false
47
+ use_ema: false
48
+ val_image_url: null
49
+ validation_epochs: 1
50
+ validation_prompt: null
51
+ variant: null
logs/instruct-pix2pix/1739029905.340061/events.out.tfevents.1739029905.spcc-a40g14.473747.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fcacb0cc69f57f5c8b24f4a4a19d1a42cd30cdd0aa205e589897940630a1332a
3
+ size 2448
logs/instruct-pix2pix/1739029905.3451133/hparams.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 5000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: null
14
+ edit_prompt_column: instruction
15
+ edited_image_column: edited_image
16
+ enable_xformers_memory_efficient_attention: false
17
+ gradient_accumulation_steps: 4
18
+ gradient_checkpointing: true
19
+ hub_model_id: null
20
+ hub_token: null
21
+ learning_rate: 5.0e-05
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 0
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 15000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 334
32
+ num_validation_images: 4
33
+ original_image_column: image_name
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: stable-diffusion-v1-5/stable-diffusion-v1-5
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 256
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 4
45
+ train_data_dir: ./self_barefoot_dataset/new_structure/
46
+ use_8bit_adam: false
47
+ use_ema: false
48
+ val_image_url: null
49
+ validation_epochs: 1
50
+ validation_prompt: null
51
+ variant: null
logs/instruct-pix2pix/1739030361.8407564/events.out.tfevents.1739030361.spcc-a40g14.476260.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:399871eec6d60ea99518284915baa76bf67c35aee8097f1314bdd71e05009bc7
3
+ size 2448
logs/instruct-pix2pix/1739030361.8500128/hparams.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 5000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: null
14
+ edit_prompt_column: instruction
15
+ edited_image_column: edited_image
16
+ enable_xformers_memory_efficient_attention: false
17
+ gradient_accumulation_steps: 4
18
+ gradient_checkpointing: true
19
+ hub_model_id: null
20
+ hub_token: null
21
+ learning_rate: 5.0e-05
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 0
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 15000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 334
32
+ num_validation_images: 4
33
+ original_image_column: image_name
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: stable-diffusion-v1-5/stable-diffusion-v1-5
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 256
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 4
45
+ train_data_dir: ./self_barefoot_dataset/new_structure/
46
+ use_8bit_adam: false
47
+ use_ema: false
48
+ val_image_url: null
49
+ validation_epochs: 1
50
+ validation_prompt: null
51
+ variant: null
logs/instruct-pix2pix/1739030438.2365324/events.out.tfevents.1739030438.spcc-a40g14.476772.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4697b4e75b92abc69809bd2f2de0ad6eca8a9ff1503a631d51684d88754777fd
3
+ size 2448
logs/instruct-pix2pix/1739030438.2423885/hparams.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 5000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: null
14
+ edit_prompt_column: instruction
15
+ edited_image_column: edited_image
16
+ enable_xformers_memory_efficient_attention: false
17
+ gradient_accumulation_steps: 4
18
+ gradient_checkpointing: true
19
+ hub_model_id: null
20
+ hub_token: null
21
+ learning_rate: 5.0e-05
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 0
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 15000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 334
32
+ num_validation_images: 4
33
+ original_image_column: image_name
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: stable-diffusion-v1-5/stable-diffusion-v1-5
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 256
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 4
45
+ train_data_dir: ./self_barefoot_dataset/new_structure/
46
+ use_8bit_adam: false
47
+ use_ema: false
48
+ val_image_url: null
49
+ validation_epochs: 1
50
+ validation_prompt: null
51
+ variant: null
logs/instruct-pix2pix/1739030577.7740219/events.out.tfevents.1739030577.spcc-a40g14.477553.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76c027ead7f9fcef951dbf888ba2bb5e98db0fae2e53b440eff4096f9f407c0a
3
+ size 2448
logs/instruct-pix2pix/1739030577.7788777/hparams.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 5000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: null
14
+ edit_prompt_column: instruction
15
+ edited_image_column: edited_image
16
+ enable_xformers_memory_efficient_attention: false
17
+ gradient_accumulation_steps: 8
18
+ gradient_checkpointing: true
19
+ hub_model_id: null
20
+ hub_token: null
21
+ learning_rate: 5.0e-05
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 0
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 15000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 1250
32
+ num_validation_images: 4
33
+ original_image_column: image_name
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: stable-diffusion-v1-5/stable-diffusion-v1-5
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 8
45
+ train_data_dir: ./self_barefoot_dataset/new_structure/
46
+ use_8bit_adam: false
47
+ use_ema: false
48
+ val_image_url: null
49
+ validation_epochs: 1
50
+ validation_prompt: null
51
+ variant: null
logs/instruct-pix2pix/1739030713.9233325/events.out.tfevents.1739030713.spcc-a40g14.478386.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a65cc0b1215bbe69414fe9d6dcd457397014910f77b372f3aa25056cfc0b22d5
3
+ size 2448
logs/instruct-pix2pix/1739030713.9285104/hparams.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 5000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: null
14
+ edit_prompt_column: instruction
15
+ edited_image_column: edited_image
16
+ enable_xformers_memory_efficient_attention: false
17
+ gradient_accumulation_steps: 4
18
+ gradient_checkpointing: true
19
+ hub_model_id: null
20
+ hub_token: null
21
+ learning_rate: 5.0e-05
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 0
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 15000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 334
32
+ num_validation_images: 4
33
+ original_image_column: image_name
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: stable-diffusion-v1-5/stable-diffusion-v1-5
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 1024
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 4
45
+ train_data_dir: ./self_barefoot_dataset/new_structure/
46
+ use_8bit_adam: false
47
+ use_ema: false
48
+ val_image_url: null
49
+ validation_epochs: 1
50
+ validation_prompt: null
51
+ variant: null
logs/instruct-pix2pix/1739031020.4435735/events.out.tfevents.1739031020.spcc-a40g14.479886.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa420538c6ba01e8efa22c164ba0f39db04d1bb1ec1d50e9dae5bf41f03a788c
3
+ size 2448
logs/instruct-pix2pix/1739031020.4503279/hparams.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 200
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: null
14
+ edit_prompt_column: instruction
15
+ edited_image_column: edited_image
16
+ enable_xformers_memory_efficient_attention: false
17
+ gradient_accumulation_steps: 4
18
+ gradient_checkpointing: true
19
+ hub_model_id: null
20
+ hub_token: null
21
+ learning_rate: 5.0e-05
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 0
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 2000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 45
32
+ num_validation_images: 4
33
+ original_image_column: image_name
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: stable-diffusion-v1-5/stable-diffusion-v1-5
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 1024
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 4
45
+ train_data_dir: ./self_barefoot_dataset/new_structure/
46
+ use_8bit_adam: false
47
+ use_ema: false
48
+ val_image_url: null
49
+ validation_epochs: 1
50
+ validation_prompt: null
51
+ variant: null
logs/instruct-pix2pix/events.out.tfevents.1739018978.spcc-a40g19.1239308.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e7a8ad11fc7d627e19a6eb9a1091869cfbbb3f7f37ec881ea8ee9fb34aae272c
3
+ size 94972
logs/instruct-pix2pix/events.out.tfevents.1739025279.spcc-a40g14.448688.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9800322290748eaba5160857753d8c3c75324c8a0f7dfe2fbc2cbf70bdbbd106
3
+ size 60280
logs/instruct-pix2pix/events.out.tfevents.1739029644.spcc-a40g14.472330.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5e020f087b1eee95833bb81bd730049535bd6c2714aa1cbd98c750fe25ce7b25
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1739029905.spcc-a40g14.473747.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:64cd014b5f2d71a2642f6a5f50c73fcdb483282bd2164e9a769007e55141fefa
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1739030361.spcc-a40g14.476260.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f91bd85c24885fb553c96193a5743bf72d0dfd95514a401f6712032936f1de0e
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1739030438.spcc-a40g14.476772.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:75268b871a0e971dc50d05b0719b44c9eb795754c9c7132b9d2eb53e35b21cbd
3
+ size 664
logs/instruct-pix2pix/events.out.tfevents.1739030577.spcc-a40g14.477553.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:556fba99bd22505e524f8cea8bbe197d2fea50fdff78d665fc053032b59765f7
3
+ size 232
logs/instruct-pix2pix/events.out.tfevents.1739030713.spcc-a40g14.478386.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:02a8a9b49e5eb9763cf9fde7a212ed09f834add74cf00f178f8942fb27472406
3
+ size 856
logs/instruct-pix2pix/events.out.tfevents.1739031020.spcc-a40g14.479886.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e9f5509d1c16d2ea919f835d11abc65971825aae13ca515e5022a8a9c08d6ea0
3
+ size 97961
model_index.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "StableDiffusionInstructPix2PixPipeline",
3
+ "_diffusers_version": "0.33.0.dev0",
4
+ "_name_or_path": "stable-diffusion-v1-5/stable-diffusion-v1-5",
5
+ "feature_extractor": [
6
+ "transformers",
7
+ "CLIPImageProcessor"
8
+ ],
9
+ "image_encoder": [
10
+ null,
11
+ null
12
+ ],
13
+ "requires_safety_checker": true,
14
+ "safety_checker": [
15
+ "stable_diffusion",
16
+ "StableDiffusionSafetyChecker"
17
+ ],
18
+ "scheduler": [
19
+ "diffusers",
20
+ "PNDMScheduler"
21
+ ],
22
+ "text_encoder": [
23
+ "transformers",
24
+ "CLIPTextModel"
25
+ ],
26
+ "tokenizer": [
27
+ "transformers",
28
+ "CLIPTokenizer"
29
+ ],
30
+ "unet": [
31
+ "diffusers",
32
+ "UNet2DConditionModel"
33
+ ],
34
+ "vae": [
35
+ "diffusers",
36
+ "AutoencoderKL"
37
+ ]
38
+ }
safety_checker/config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "./my_huggingface_cache/hub/models--stable-diffusion-v1-5--stable-diffusion-v1-5/snapshots/451f4fe16113bff5a5d2269ed5ad43b0592e9a14/safety_checker",
3
+ "architectures": [
4
+ "StableDiffusionSafetyChecker"
5
+ ],
6
+ "initializer_factor": 1.0,
7
+ "logit_scale_init_value": 2.6592,
8
+ "model_type": "clip",
9
+ "projection_dim": 768,
10
+ "text_config": {
11
+ "dropout": 0.0,
12
+ "hidden_size": 768,
13
+ "intermediate_size": 3072,
14
+ "model_type": "clip_text_model",
15
+ "num_attention_heads": 12
16
+ },
17
+ "torch_dtype": "float32",
18
+ "transformers_version": "4.46.3",
19
+ "vision_config": {
20
+ "_attn_implementation_autoset": true,
21
+ "dropout": 0.0,
22
+ "hidden_size": 1024,
23
+ "intermediate_size": 4096,
24
+ "model_type": "clip_vision_model",
25
+ "num_attention_heads": 16,
26
+ "num_hidden_layers": 24,
27
+ "patch_size": 14
28
+ }
29
+ }
safety_checker/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb351a5ded815c3ff744968ad9c6b218d071b9d313d04f35e813b84b4c0ffde8
3
+ size 1215979664
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "PNDMScheduler",
3
+ "_diffusers_version": "0.33.0.dev0",
4
+ "beta_end": 0.012,
5
+ "beta_schedule": "scaled_linear",
6
+ "beta_start": 0.00085,
7
+ "clip_sample": false,
8
+ "num_train_timesteps": 1000,
9
+ "prediction_type": "epsilon",
10
+ "set_alpha_to_one": false,
11
+ "skip_prk_steps": true,
12
+ "steps_offset": 1,
13
+ "timestep_spacing": "leading",
14
+ "trained_betas": null
15
+ }
text_encoder/config.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "stable-diffusion-v1-5/stable-diffusion-v1-5",
3
+ "architectures": [
4
+ "CLIPTextModel"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 0,
8
+ "dropout": 0.0,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "quick_gelu",
11
+ "hidden_size": 768,
12
+ "initializer_factor": 1.0,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 77,
17
+ "model_type": "clip_text_model",
18
+ "num_attention_heads": 12,
19
+ "num_hidden_layers": 12,
20
+ "pad_token_id": 1,
21
+ "projection_dim": 768,
22
+ "torch_dtype": "float16",
23
+ "transformers_version": "4.46.3",
24
+ "vocab_size": 49408
25
+ }
text_encoder/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:660c6f5b1abae9dc498ac2d21e1347d2abdb0cf6c0c0c8576cd796491d9a6cdd
3
+ size 246144152
tokenizer/merges.txt ADDED
The diff for this file is too large to render. See raw diff