hhd23 commited on
Commit
c6380c9
1 Parent(s): 5e6a197

End of training

Browse files
checkpoint-18000/optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e4c2352c780637debbf717d2b2de0b43b322b07b56da1d1d1fa374bbad302d27
3
- size 6876851924
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2475f167bfe200a48870cfd02004388e8f2e38dcca5521647dd7b93dd3642cda
3
+ size 6876842324
checkpoint-18000/random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4363168584660c7c735cb50d7571d6b6ea6711552bed9a26288ac51f7ea9419e
3
  size 14604
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d490ca8c5265906f213c4119a65f93ca7f06023aec8f78e16cc1fb5794ece124
3
  size 14604
checkpoint-18000/scaler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:70e8b031e2534fc6a1aa73020f3481942036cc743bdd5097cfbda008e32c98c5
3
  size 988
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cda8f4489661a5cbcb7fe5a81efde2d932989f2d13f409fe336cdd55a90d6d58
3
  size 988
checkpoint-18000/unet/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.28.0.dev0",
4
- "_name_or_path": "instruct-pix2pix-model/checkpoint-9000",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
7
  "addition_embed_type_num_heads": 64,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.28.0.dev0",
4
+ "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
7
  "addition_embed_type_num_heads": 64,
checkpoint-18000/unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3552362f5945eba3f8c1c24a7b6b8f5939a2604ed079a2aaba180deaab3d2446
3
  size 3438213624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eaced49856baafff0503af539e01f3f86a745a5fc0fdd97be0c9106759b98aef
3
  size 3438213624
logs/instruct-pix2pix/1713345561.1454492/events.out.tfevents.1713345561.ss7049i.40685.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c95fa06ad70a577269e8f18cd378abf30275e17233e967dab3d26ef4e1a46e58
3
+ size 2394
logs/instruct-pix2pix/1713345561.147853/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: 3000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: ''
14
+ edit_prompt_column: edit_prompt
15
+ edited_image_column: edited_image
16
+ enable_xformers_memory_efficient_attention: true
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: 20000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 36
32
+ num_validation_images: 4
33
+ original_image_column: input_image
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: runwayml/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/1713345603.276569/events.out.tfevents.1713345603.ss7049i.41421.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a5a2349863a95b1ddd8011b1a48bb2d9a5373ee1f01b3d66743a99d102f9b9c4
3
+ size 2394
logs/instruct-pix2pix/1713345603.2791326/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: 3000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: ''
14
+ edit_prompt_column: edit_prompt
15
+ edited_image_column: edited_image
16
+ enable_xformers_memory_efficient_attention: true
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: 20000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 36
32
+ num_validation_images: 4
33
+ original_image_column: input_image
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: runwayml/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/events.out.tfevents.1713345561.ss7049i.40685.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e1554880f3623b901c2c4a1363b4d45cf440b3c170719a77d6d33fd4e3202a8a
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1713345603.ss7049i.41421.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34060e83258ee8ddcc5ed3268276bcdf735fe28a80efb3eabe4eb73fd23915d6
3
+ size 983578
unet/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.28.0.dev0",
4
- "_name_or_path": "instruct-pix2pix-model/checkpoint-9000",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
7
  "addition_embed_type_num_heads": 64,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.28.0.dev0",
4
+ "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
7
  "addition_embed_type_num_heads": 64,
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e0f29ad3b63bcaa7a49ae57c514e76e095a8e033a4ff9458c4b5f25c429205de
3
  size 3438213624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c0dc297976b98831c5648ba752ed7db89dfaa4f64209725ccc798012e3bf4f0
3
  size 3438213624