Stevenzhanshi commited on
Commit
c5a1e1a
·
verified ·
1 Parent(s): fe14cdc

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-1000/optimizer.bin +3 -0
  2. checkpoint-1000/random_states_0.pkl +3 -0
  3. checkpoint-1000/scaler.pt +3 -0
  4. checkpoint-1000/scheduler.bin +3 -0
  5. checkpoint-1000/unet/config.json +68 -0
  6. checkpoint-1000/unet/diffusion_pytorch_model.safetensors +3 -0
  7. feature_extractor/preprocessor_config.json +27 -0
  8. logs/instruct-pix2pix/1744991822.9506016/events.out.tfevents.1744991822.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.2303.1 +3 -0
  9. logs/instruct-pix2pix/1744991822.9843452/hparams.yml +51 -0
  10. logs/instruct-pix2pix/1745051623.3535006/events.out.tfevents.1745051623.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.2908.1 +3 -0
  11. logs/instruct-pix2pix/1745051623.3786075/hparams.yml +51 -0
  12. logs/instruct-pix2pix/1745051880.0799575/events.out.tfevents.1745051880.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3058.1 +3 -0
  13. logs/instruct-pix2pix/1745051880.1131375/hparams.yml +51 -0
  14. logs/instruct-pix2pix/1745051978.1224022/events.out.tfevents.1745051978.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3208.1 +3 -0
  15. logs/instruct-pix2pix/1745051978.149768/hparams.yml +51 -0
  16. logs/instruct-pix2pix/1745052698.949806/events.out.tfevents.1745052698.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3358.1 +3 -0
  17. logs/instruct-pix2pix/1745052698.9776397/hparams.yml +51 -0
  18. logs/instruct-pix2pix/1745052760.7454703/events.out.tfevents.1745052760.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3508.1 +3 -0
  19. logs/instruct-pix2pix/1745052760.7726707/hparams.yml +51 -0
  20. logs/instruct-pix2pix/1745053073.8717396/events.out.tfevents.1745053073.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3658.1 +3 -0
  21. logs/instruct-pix2pix/1745053073.8990517/hparams.yml +51 -0
  22. logs/instruct-pix2pix/1745053235.5696728/events.out.tfevents.1745053235.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3808.1 +3 -0
  23. logs/instruct-pix2pix/1745053235.59478/hparams.yml +51 -0
  24. logs/instruct-pix2pix/1745053296.8217793/events.out.tfevents.1745053296.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3958.1 +3 -0
  25. logs/instruct-pix2pix/1745053296.8500984/hparams.yml +51 -0
  26. logs/instruct-pix2pix/1745496150.8769064/events.out.tfevents.1745496150.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-2s70vh-worker.2059.1 +3 -0
  27. logs/instruct-pix2pix/1745496150.9082656/hparams.yml +51 -0
  28. logs/instruct-pix2pix/1745502128.123976/events.out.tfevents.1745502128.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-2s70vh-worker.2776.1 +3 -0
  29. logs/instruct-pix2pix/1745502128.1537488/hparams.yml +51 -0
  30. logs/instruct-pix2pix/1745502541.4668396/events.out.tfevents.1745502541.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-2s70vh-worker.2998.1 +3 -0
  31. logs/instruct-pix2pix/1745502541.4967284/hparams.yml +51 -0
  32. logs/instruct-pix2pix/events.out.tfevents.1744991822.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.2303.0 +3 -0
  33. logs/instruct-pix2pix/events.out.tfevents.1745051623.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.2908.0 +3 -0
  34. logs/instruct-pix2pix/events.out.tfevents.1745051880.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3058.0 +3 -0
  35. logs/instruct-pix2pix/events.out.tfevents.1745051978.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3208.0 +3 -0
  36. logs/instruct-pix2pix/events.out.tfevents.1745052698.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3358.0 +3 -0
  37. logs/instruct-pix2pix/events.out.tfevents.1745052760.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3508.0 +3 -0
  38. logs/instruct-pix2pix/events.out.tfevents.1745053073.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3658.0 +3 -0
  39. logs/instruct-pix2pix/events.out.tfevents.1745053235.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3808.0 +3 -0
  40. logs/instruct-pix2pix/events.out.tfevents.1745053296.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3958.0 +3 -0
  41. logs/instruct-pix2pix/events.out.tfevents.1745496150.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-2s70vh-worker.2059.0 +3 -0
  42. logs/instruct-pix2pix/events.out.tfevents.1745502128.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-2s70vh-worker.2776.0 +3 -0
  43. logs/instruct-pix2pix/events.out.tfevents.1745502541.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-2s70vh-worker.2998.0 +3 -0
  44. model_index.json +38 -0
  45. safety_checker/config.json +47 -0
  46. safety_checker/model.safetensors +3 -0
  47. scheduler/scheduler_config.json +23 -0
  48. text_encoder/config.json +24 -0
  49. text_encoder/model.safetensors +3 -0
  50. tokenizer/merges.txt +0 -0
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89e78d0844c422a0404444ef4848ed36f83eddd018130d571a4f48cdb8ce7329
3
+ size 6876842324
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:030f590484d5e72364759fc18832e5caff073c7ae0c071dec66284ff965cd7f9
3
+ size 16100
checkpoint-1000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d8fdcd0311eba9854fff738038ed4c1a269832665b4d88ba4e4e3d02a1a7e0e
3
+ size 988
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d16ae9dedf4c45ceadc467593037d83423677715e75ab402a9f44aa81de9b84
3
+ size 1000
checkpoint-1000/unet/config.json ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.34.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,
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-1000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8e5c65e6affc479dfe006024306e169d820fbd14f068a0f50bb9201dc4769da
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/1744991822.9506016/events.out.tfevents.1744991822.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.2303.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aea84d75e52fa234ff7f38d394f75eeda0240f7077ab0e8b360bd14b430f6910
3
+ size 2418
logs/instruct-pix2pix/1744991822.9843452/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: 25
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.1
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: youlive789/instructpix2pix
14
+ edit_prompt_column: edit_promt
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: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 20
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 100
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 1
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: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 2
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/1745051623.3535006/events.out.tfevents.1745051623.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.2908.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbf144d01d471abef647b412e6d37023cc07b232f4a8f8785cb7a532cffb236e
3
+ size 2400
logs/instruct-pix2pix/1745051623.3786075/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: 25
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.1
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: Aurora1609/Pix2Pix_RoboTwin
14
+ edit_prompt_column: prompt
15
+ edited_image_column: after
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: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 20
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 100
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 4
33
+ original_image_column: before
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: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 1
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/1745051880.0799575/events.out.tfevents.1745051880.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3058.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a72c4b5007ad8e45f1a0b4995e7fe32d4fb5c317bedeb4a2b4d657a62777172d
3
+ size 2400
logs/instruct-pix2pix/1745051880.1131375/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: 25
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.1
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: Aurora1609/Pix2Pix_RoboTwin
14
+ edit_prompt_column: prompt
15
+ edited_image_column: after
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: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 20
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 100
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 4
33
+ original_image_column: before
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: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 1
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/1745051978.1224022/events.out.tfevents.1745051978.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3208.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ab9cb4d1544674a8ad6d91b4520ec2193eb9cec434ebe43558e72f334ddaab9
3
+ size 2400
logs/instruct-pix2pix/1745051978.149768/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: 25
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.1
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: Aurora1609/Pix2Pix_RoboTwin
14
+ edit_prompt_column: prompt
15
+ edited_image_column: after
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: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 20
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 100
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 4
33
+ original_image_column: before
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: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 1
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/1745052698.949806/events.out.tfevents.1745052698.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3358.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a4ddb14a88de5cbb5128556978eec82784f37279263faff88c480115e46ac7d
3
+ size 2400
logs/instruct-pix2pix/1745052698.9776397/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: 25
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.1
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: Aurora1609/Pix2Pix_RoboTwin
14
+ edit_prompt_column: prompt
15
+ edited_image_column: after
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: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 20
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 100
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 4
33
+ original_image_column: before
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: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 1
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/1745052760.7454703/events.out.tfevents.1745052760.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3508.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad92d3a21b861f847ddea8957d7f1db84bcdf16b8a76c5eaed4fb0e51bbb9c0a
3
+ size 2400
logs/instruct-pix2pix/1745052760.7726707/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: 25
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.1
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: Aurora1609/Pix2Pix_RoboTwin
14
+ edit_prompt_column: prompt
15
+ edited_image_column: after
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: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 20
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 100
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 4
33
+ original_image_column: before
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: 128
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 1
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/1745053073.8717396/events.out.tfevents.1745053073.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3658.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c5722ed54c98e7df3891fe7c3f89defeb6431fffad0a69e4b6cb092209416135
3
+ size 2400
logs/instruct-pix2pix/1745053073.8990517/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: 25
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.1
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: Aurora1609/Pix2Pix_RoboTwin
14
+ edit_prompt_column: prompt
15
+ edited_image_column: after
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: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 20
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 100
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 4
33
+ original_image_column: before
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: 128
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 1
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/1745053235.5696728/events.out.tfevents.1745053235.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3808.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88f141795bd9cabf366581fc205681a92de1079192b43817a12eaa2ceffb008b
3
+ size 2400
logs/instruct-pix2pix/1745053235.59478/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: 25
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.1
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: Aurora1609/Pix2Pix_RoboTwin
14
+ edit_prompt_column: prompt
15
+ edited_image_column: after
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: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 20
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 100
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 4
33
+ original_image_column: before
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: 128
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 1
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/1745053296.8217793/events.out.tfevents.1745053296.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3958.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:000efa8998899db04e842e2087410f72478c972be099fcdb46d9650ff0c870ad
3
+ size 2400
logs/instruct-pix2pix/1745053296.8500984/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: 25
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.1
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: Aurora1609/Pix2Pix_RoboTwin
14
+ edit_prompt_column: prompt
15
+ edited_image_column: after
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: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 20
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 100
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 4
33
+ original_image_column: before
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: 128
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 1
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/1745496150.8769064/events.out.tfevents.1745496150.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-2s70vh-worker.2059.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9cc5eb5b562efe1e91046970d138acce562a470db183741f91a3e08ac87aa2e6
3
+ size 2392
logs/instruct-pix2pix/1745496150.9082656/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: 25
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.1
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: Aurora1609/RoboTwin
14
+ edit_prompt_column: prompt
15
+ edited_image_column: after
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: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 20
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 100
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 4
33
+ original_image_column: before
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: 128
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 1
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/1745502128.123976/events.out.tfevents.1745502128.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-2s70vh-worker.2776.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f235b325a9829a381a0b7625f6f907765f10f7d04f090f030450d2d763d9d93
3
+ size 2392
logs/instruct-pix2pix/1745502128.1537488/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: 25
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.1
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: Aurora1609/RoboTwin
14
+ edit_prompt_column: prompt
15
+ edited_image_column: after
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: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 20
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 1000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 4
33
+ original_image_column: before
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: 128
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 1
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/1745502541.4668396/events.out.tfevents.1745502541.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-2s70vh-worker.2998.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c80b733a79ff47dd9686cc2c49927d85d00d7cbc14fc32af3657f9d258ad12cd
3
+ size 2392
logs/instruct-pix2pix/1745502541.4967284/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: 25
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.1
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: Aurora1609/RoboTwin
14
+ edit_prompt_column: prompt
15
+ edited_image_column: after
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: 1.0e-05
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 20
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 1000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 4
33
+ original_image_column: before
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: 128
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 1
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.1744991822.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.2303.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:65e397b8caadf48216cd6d33d6766165497f79684ac80721259fc896c4c11230
3
+ size 4888
logs/instruct-pix2pix/events.out.tfevents.1745051623.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.2908.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:78c55f4d344c15ee50ce79d75c9905248fee514491f211397ef8d882810f7f4a
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1745051880.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3058.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:732f2383b44c2c58c77b7431e46c93c7fdc7be0b6be378c107c6ab359a5848a2
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1745051978.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3208.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e85b7be0e8c86327a3e0636d628dee637b7fd0ccf65038959cef962242c7b56a
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1745052698.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3358.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4ef416fe5c0de38224619176066a32f49909cff1f80034a34a9daf035c9139d
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1745052760.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3508.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3348f03f1bda03c611f9e50d7a7419463b0d70e4c34d7801a97e0f0b7cc0f7e7
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1745053073.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3658.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b3749eebdbfad215b5d2849fe2d37d1b3f5b189be053d1230c76aa528046b18b
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1745053235.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3808.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:00ce63cf974f7ed5bce7b2fca9ba2c195381cdf7c10584e41851488cda9e1f94
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1745053296.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-p403wp-worker.3958.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b19b6e5e2f9cd9a2154fb39322c57f337b8eb1a58016809accb7f0ea1454395
3
+ size 4888
logs/instruct-pix2pix/events.out.tfevents.1745496150.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-2s70vh-worker.2059.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b71dea0774f1b9a081e5368ee8c2033cf58e69119034755d4d9c1b024cb17ebc
3
+ size 4888
logs/instruct-pix2pix/events.out.tfevents.1745502128.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-2s70vh-worker.2776.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61353990d458260e4a2e6f8568b741cc8171bfa47694284cf98de70fc93647ce
3
+ size 6674
logs/instruct-pix2pix/events.out.tfevents.1745502541.mlxlabzvy78jw467d8df06-20250318024839-7a3w86-2s70vh-worker.2998.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d3ab44c8ba9bfc255b698c2ea89a6755473f0d9be5b2b9eb91efd36fbc8819d
3
+ size 48961
model_index.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "StableDiffusionInstructPix2PixPipeline",
3
+ "_diffusers_version": "0.34.0.dev0",
4
+ "_name_or_path": "timbrooks/instruct-pix2pix",
5
+ "feature_extractor": [
6
+ "transformers",
7
+ "CLIPImageProcessor"
8
+ ],
9
+ "image_encoder": [
10
+ null,
11
+ null
12
+ ],
13
+ "requires_safety_checker": false,
14
+ "safety_checker": [
15
+ "stable_diffusion",
16
+ "StableDiffusionSafetyChecker"
17
+ ],
18
+ "scheduler": [
19
+ "diffusers",
20
+ "EulerAncestralDiscreteScheduler"
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,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "StableDiffusionSafetyChecker"
4
+ ],
5
+ "initializer_factor": 1.0,
6
+ "logit_scale_init_value": 2.6592,
7
+ "model_type": "clip",
8
+ "projection_dim": 768,
9
+ "text_config": {
10
+ "attention_dropout": 0.0,
11
+ "dropout": 0.0,
12
+ "hidden_act": "quick_gelu",
13
+ "hidden_size": 768,
14
+ "initializer_factor": 1.0,
15
+ "initializer_range": 0.02,
16
+ "intermediate_size": 3072,
17
+ "layer_norm_eps": 1e-05,
18
+ "max_position_embeddings": 77,
19
+ "model_type": "clip_text_model",
20
+ "num_attention_heads": 12,
21
+ "num_hidden_layers": 12,
22
+ "projection_dim": 512,
23
+ "torch_dtype": "float32",
24
+ "vocab_size": 49408
25
+ },
26
+ "torch_dtype": "float32",
27
+ "transformers_version": "4.51.3",
28
+ "vision_config": {
29
+ "_attn_implementation_autoset": true,
30
+ "attention_dropout": 0.0,
31
+ "dropout": 0.0,
32
+ "hidden_act": "quick_gelu",
33
+ "hidden_size": 1024,
34
+ "image_size": 224,
35
+ "initializer_factor": 1.0,
36
+ "initializer_range": 0.02,
37
+ "intermediate_size": 4096,
38
+ "layer_norm_eps": 1e-05,
39
+ "model_type": "clip_vision_model",
40
+ "num_attention_heads": 16,
41
+ "num_channels": 3,
42
+ "num_hidden_layers": 24,
43
+ "patch_size": 14,
44
+ "projection_dim": 512,
45
+ "torch_dtype": "float32"
46
+ }
47
+ }
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,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "EulerAncestralDiscreteScheduler",
3
+ "_diffusers_version": "0.34.0.dev0",
4
+ "algorithm_type": "dpmsolver++",
5
+ "beta_end": 0.012,
6
+ "beta_schedule": "scaled_linear",
7
+ "beta_start": 0.00085,
8
+ "clip_sample": false,
9
+ "dynamic_thresholding_ratio": 0.995,
10
+ "lower_order_final": true,
11
+ "num_train_timesteps": 1000,
12
+ "prediction_type": "epsilon",
13
+ "rescale_betas_zero_snr": false,
14
+ "sample_max_value": 1.0,
15
+ "set_alpha_to_one": false,
16
+ "skip_prk_steps": true,
17
+ "solver_order": 2,
18
+ "solver_type": "midpoint",
19
+ "steps_offset": 1,
20
+ "thresholding": false,
21
+ "timestep_spacing": "linspace",
22
+ "trained_betas": null
23
+ }
text_encoder/config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "CLIPTextModel"
4
+ ],
5
+ "attention_dropout": 0.0,
6
+ "bos_token_id": 0,
7
+ "dropout": 0.0,
8
+ "eos_token_id": 2,
9
+ "hidden_act": "quick_gelu",
10
+ "hidden_size": 768,
11
+ "initializer_factor": 1.0,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 3072,
14
+ "layer_norm_eps": 1e-05,
15
+ "max_position_embeddings": 77,
16
+ "model_type": "clip_text_model",
17
+ "num_attention_heads": 12,
18
+ "num_hidden_layers": 12,
19
+ "pad_token_id": 1,
20
+ "projection_dim": 768,
21
+ "torch_dtype": "float16",
22
+ "transformers_version": "4.51.3",
23
+ "vocab_size": 49408
24
+ }
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