Hugging Face's logo

Stevenzhanshi
/
instruct-pix2pix-model

Diffusers
TensorBoard
Safetensors
StableDiffusionInstructPix2PixPipeline
Model card Files
xet
Metrics Community
Stevenzhanshi commited on
Commit
f25a090
·
verified ·
1 Parent(s): f32d53d

End of training

Browse files
Files changed (28) hide show
  1. checkpoint-5000/optimizer.bin +3 -0
  2. checkpoint-5000/random_states_0.pkl +3 -0
  3. checkpoint-5000/scaler.pt +3 -0
  4. checkpoint-5000/scheduler.bin +3 -0
  5. checkpoint-5000/unet/config.json +68 -0
  6. checkpoint-5000/unet/diffusion_pytorch_model.safetensors +3 -0
  7. logs/instruct-pix2pix/1746289878.1617947/events.out.tfevents.1746289878.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.348.1 +3 -0
  8. logs/instruct-pix2pix/1746289878.1904514/hparams.yml +51 -0
  9. logs/instruct-pix2pix/1746293446.5591884/events.out.tfevents.1746293446.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.665.1 +3 -0
  10. logs/instruct-pix2pix/1746293446.5905352/hparams.yml +51 -0
  11. logs/instruct-pix2pix/1746293855.7252355/events.out.tfevents.1746293855.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.964.1 +3 -0
  12. logs/instruct-pix2pix/1746293855.7634168/hparams.yml +51 -0
  13. logs/instruct-pix2pix/1746330705.9422994/events.out.tfevents.1746330705.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.1117.1 +3 -0
  14. logs/instruct-pix2pix/1746330705.9715006/hparams.yml +51 -0
  15. logs/instruct-pix2pix/1746331598.4573915/events.out.tfevents.1746331598.mlxlab115rk4ej68007254-20250417031533-w85uwi-38v0sa-worker.710.1 +3 -0
  16. logs/instruct-pix2pix/1746331598.4933681/hparams.yml +51 -0
  17. logs/instruct-pix2pix/1746333719.959134/events.out.tfevents.1746333719.mlxlab115rk4ej68007254-20250417031533-w85uwi-qceq6t-worker.245.1 +3 -0
  18. logs/instruct-pix2pix/1746333719.9885886/hparams.yml +51 -0
  19. logs/instruct-pix2pix/1746333782.5425582/events.out.tfevents.1746333782.mlxlab115rk4ej68007254-20250417031533-w85uwi-qceq6t-worker.549.1 +3 -0
  20. logs/instruct-pix2pix/1746333782.5719867/hparams.yml +51 -0
  21. logs/instruct-pix2pix/events.out.tfevents.1746289878.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.348.0 +3 -0
  22. logs/instruct-pix2pix/events.out.tfevents.1746293446.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.665.0 +3 -0
  23. logs/instruct-pix2pix/events.out.tfevents.1746293855.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.964.0 +3 -0
  24. logs/instruct-pix2pix/events.out.tfevents.1746330705.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.1117.0 +3 -0
  25. logs/instruct-pix2pix/events.out.tfevents.1746331598.mlxlab115rk4ej68007254-20250417031533-w85uwi-38v0sa-worker.710.0 +3 -0
  26. logs/instruct-pix2pix/events.out.tfevents.1746333719.mlxlab115rk4ej68007254-20250417031533-w85uwi-qceq6t-worker.245.0 +3 -0
  27. logs/instruct-pix2pix/events.out.tfevents.1746333782.mlxlab115rk4ej68007254-20250417031533-w85uwi-qceq6t-worker.549.0 +3 -0
  28. unet/diffusion_pytorch_model.safetensors +1 -1
checkpoint-5000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f39780ed680dd9bc3244db786a49bd569ddf39db16002ffe0cc072b2f6dcb0c5
3
+ size 6876849433
checkpoint-5000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d4b612526ae292312568b73854896a87c6377b40d2cb8e7e55ef93c0db8d0295
3
+ size 14821
checkpoint-5000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82ad8990572ad11a824b7db276c8af49c179ca7e7724b4e6906cd0ae480a80a8
3
+ size 1383
checkpoint-5000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e406666b98a93a8a4939cbac8b855696fbc758b8c913254c77275a0aa59eb06c
3
+ size 1401
checkpoint-5000/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-5000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c704272af80dc6b149b01830c7db4cf39479679c809b67f41a6937c470c47357
3
+ size 3438213624
logs/instruct-pix2pix/1746289878.1617947/events.out.tfevents.1746289878.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.348.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a714b1775b887483894228d36a802b98d1c9797aa5bfb7e4ee8e9865b8dc03f0
3
+ size 2392
logs/instruct-pix2pix/1746289878.1904514/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: true
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 500
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: 2000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 2
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: false
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/1746293446.5591884/events.out.tfevents.1746293446.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.665.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f317b36b23f535784e5827eb70290074fecf91a70c624aaeb5f8b099f07876f2
3
+ size 2392
logs/instruct-pix2pix/1746293446.5905352/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: true
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 500
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: 2000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 14
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: false
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: 8
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/1746293855.7252355/events.out.tfevents.1746293855.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.964.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e58920e6e06f454e1435d0bb51fa272a54ad98f9a0b352d575d0cd640327f3b
3
+ size 2392
logs/instruct-pix2pix/1746293855.7634168/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: 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: 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: 50
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: 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/1746330705.9422994/events.out.tfevents.1746330705.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.1117.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f33751a654bb47ac531190c4d90f0ec48a6ff43a2fc079f70860764eb5fd4064
3
+ size 2392
logs/instruct-pix2pix/1746330705.9715006/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: 500
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
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: 0
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: 100
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: 128
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/1746331598.4573915/events.out.tfevents.1746331598.mlxlab115rk4ej68007254-20250417031533-w85uwi-38v0sa-worker.710.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:597638d2ff977e609370ad8132344d3865bc946a2acd026cb864206451aade91
3
+ size 2392
logs/instruct-pix2pix/1746331598.4933681/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: 500
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
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: 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: 1000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 334
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: 128
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/1746333719.959134/events.out.tfevents.1746333719.mlxlab115rk4ej68007254-20250417031533-w85uwi-qceq6t-worker.245.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f0fd625c4d6501d0b26d94973f01d4a91309fe13beb289a66448cbc219e0207
3
+ size 2392
logs/instruct-pix2pix/1746333719.9885886/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: 500
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
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: 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: 1000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 334
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: 128
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/1746333782.5425582/events.out.tfevents.1746333782.mlxlab115rk4ej68007254-20250417031533-w85uwi-qceq6t-worker.549.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8cc704b709c0ea6a8f960d3a3a469f01eff85c152e7d4c7f4d715eb185cd79b7
3
+ size 2392
logs/instruct-pix2pix/1746333782.5719867/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: 500
9
+ checkpoints_total_limit: 1
10
+ conditioning_dropout_prob: 0.05
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: 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: 100
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 34
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: 128
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.1746289878.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.348.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a6d0d7a267b79f83a17da68430c9c743b931a5ab723f8d916f05e19099987cba
3
+ size 73461
logs/instruct-pix2pix/events.out.tfevents.1746293446.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.665.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13b14790800f2fa69f3c14581b5c84b5b145202a3d3ee8a931a2527ff59e43ce
3
+ size 2056
logs/instruct-pix2pix/events.out.tfevents.1746293855.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.964.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c683b332970242dbbaa872a009abfa2483384406339718da35ad8c597f567014
3
+ size 438364
logs/instruct-pix2pix/events.out.tfevents.1746330705.mlxlab115rk4ej68007254-20250417031533-w85uwi-r77ets-worker.1117.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1fbb058ddd096f2d924e5eb6429040287434d972fe77da3f02aea60112229855
3
+ size 136
logs/instruct-pix2pix/events.out.tfevents.1746331598.mlxlab115rk4ej68007254-20250417031533-w85uwi-38v0sa-worker.710.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ca6e07f9edf60d7316915bc3d3a01fa4b0699ddd586ab465644ab1c5ac9b5ed
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1746333719.mlxlab115rk4ej68007254-20250417031533-w85uwi-qceq6t-worker.245.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4444acf84bb622ab55c8d477eeb8a2df6714300f9fc363394c835272fbeab96e
3
+ size 88
logs/instruct-pix2pix/events.out.tfevents.1746333782.mlxlab115rk4ej68007254-20250417031533-w85uwi-qceq6t-worker.549.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e5b5bc37d82137ab9698c9e2827a899930a08a205dceaffa864ae67778b67d3
3
+ size 4888
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6b69e0a5419a5ac11b4883e23fef8f703e3ed9df6b9afdd7f33db5faefe5c0d4
3
  size 3438213624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:06f55513bc39f52cd58a1e61c9b7f13f221e05938a30251d1eebf8b36045a103
3
  size 3438213624