pranjalipathre commited on
Commit
a3a52a8
1 Parent(s): eb45b35

End of training

Browse files
logs/instruct-pix2pix/1704554316.599068/events.out.tfevents.1704554316.gnode117.15391.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b749b3962d2285e71abe2495c81651f48dbc8ff6c3be6ca8cb47c7c5abb94fe
3
+ size 2424
logs/instruct-pix2pix/1704554316.603269/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: 26000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: pranjalipathre/pybullet_img2img
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: 0.0001
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 500
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 25000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 72
32
+ num_validation_images: 4
33
+ original_image_column: original_image
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: timbrooks/instruct-pix2pix
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: 40
44
+ test_image_url: null
45
+ train_batch_size: 4
46
+ train_data_dir: null
47
+ use_8bit_adam: false
48
+ use_ema: false
49
+ val_image_url: null
50
+ validation_epochs: 1
51
+ validation_prompt: null
logs/instruct-pix2pix/1704555538.1165407/events.out.tfevents.1704555538.gnode117.16408.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44f663e3c771628f3e740292bc07d274e60019a65013b5591a06c4f4fe46f8ee
3
+ size 2424
logs/instruct-pix2pix/1704555538.120376/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: 26000
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: pranjalipathre/pybullet_img2img
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: 0.0001
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 500
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 25000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 72
32
+ num_validation_images: 4
33
+ original_image_column: original_image
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: timbrooks/instruct-pix2pix
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: 40
44
+ test_image_url: null
45
+ train_batch_size: 4
46
+ train_data_dir: null
47
+ use_8bit_adam: false
48
+ use_ema: false
49
+ val_image_url: null
50
+ validation_epochs: 1
51
+ validation_prompt: null
logs/instruct-pix2pix/events.out.tfevents.1704554316.gnode117.15391.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3b5493c861c160009670f00a7dbd7281caf66961096c63ff80f9948b850f90f
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1704555538.gnode117.16408.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ffd3794fae6449c119a0a8bbee57790a27ed6672c59305f462aca1aa5e313fe9
3
+ size 1233578
model_index.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "StableDiffusionInstructPix2PixPipeline",
3
  "_diffusers_version": "0.25.0.dev0",
4
- "_name_or_path": "./instruct-pix2pix-model_150k/",
5
  "feature_extractor": [
6
  "transformers",
7
  "CLIPImageProcessor"
 
1
  {
2
  "_class_name": "StableDiffusionInstructPix2PixPipeline",
3
  "_diffusers_version": "0.25.0.dev0",
4
+ "_name_or_path": "timbrooks/instruct-pix2pix",
5
  "feature_extractor": [
6
  "transformers",
7
  "CLIPImageProcessor"
text_encoder/config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "./instruct-pix2pix-model_150k/",
3
  "architectures": [
4
  "CLIPTextModel"
5
  ],
 
1
  {
2
+ "_name_or_path": "timbrooks/instruct-pix2pix",
3
  "architectures": [
4
  "CLIPTextModel"
5
  ],
tokenizer/special_tokens_map.json CHANGED
@@ -13,13 +13,7 @@
13
  "rstrip": false,
14
  "single_word": false
15
  },
16
- "pad_token": {
17
- "content": "<|endoftext|>",
18
- "lstrip": false,
19
- "normalized": true,
20
- "rstrip": false,
21
- "single_word": false
22
- },
23
  "unk_token": {
24
  "content": "<|endoftext|>",
25
  "lstrip": false,
 
13
  "rstrip": false,
14
  "single_word": false
15
  },
16
+ "pad_token": "<|endoftext|>",
 
 
 
 
 
 
17
  "unk_token": {
18
  "content": "<|endoftext|>",
19
  "lstrip": false,
unet/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.25.0.dev0",
4
- "_name_or_path": "./instruct-pix2pix-model_150k/",
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.25.0.dev0",
4
+ "_name_or_path": "timbrooks/instruct-pix2pix",
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:5867f227f53ca54c0181ee12bc46b3c423614bbd9885640c6d9b670bfa8dbf17
3
  size 3438213624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:86d5b480016f6e65f3f7dd63c720862b989dc39cd2fc4e13dfd0393b89ba6521
3
  size 3438213624
vae/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_diffusers_version": "0.25.0.dev0",
4
- "_name_or_path": "./instruct-pix2pix-model_150k/",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_diffusers_version": "0.25.0.dev0",
4
+ "_name_or_path": "timbrooks/instruct-pix2pix",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,