dlantonia commited on
Commit
33d49a2
·
verified ·
1 Parent(s): 596a801

End of training

Browse files
checkpoint-5000/optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e19a2e1e947722a4df781aa9f27a0b06fc04d75370c1d066a8cca8e70bda7174
3
  size 6876842324
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:51a3ce546e2135e92fde77c34fde0a707c44c3f978152142bc65f9fa4286437a
3
  size 6876842324
checkpoint-5000/random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1e87cf6fa747a099bc4b7c3c6a4ed4a0c3f683f5910e34f6622d0ee21e44c29b
3
- size 14344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6933243b49a33fb967c5acfbb734aeec38889730c050a12876aac313cee88aa0
3
+ size 14408
checkpoint-5000/unet/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
- "_diffusers_version": "0.30.0.dev0",
4
  "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.33.0.dev0",
4
  "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
checkpoint-5000/unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fbabcc9cf08b34fc51fbb198755f700f45e89ed92dcfbb82545f1ff6fa5bc656
3
  size 3438213624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ccf3f72480160ea25661bfd9c48ed32352e9da426686b695207bb873d3d8c07a
3
  size 3438213624
feature_extractor/preprocessor_config.json CHANGED
@@ -1,21 +1,4 @@
1
  {
2
- "_valid_processor_keys": [
3
- "images",
4
- "do_resize",
5
- "size",
6
- "resample",
7
- "do_center_crop",
8
- "crop_size",
9
- "do_rescale",
10
- "rescale_factor",
11
- "do_normalize",
12
- "image_mean",
13
- "image_std",
14
- "do_convert_rgb",
15
- "return_tensors",
16
- "data_format",
17
- "input_data_format"
18
- ],
19
  "crop_size": {
20
  "height": 224,
21
  "width": 224
 
1
  {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  "crop_size": {
3
  "height": 224,
4
  "width": 224
logs/instruct-pix2pix/1736513441.048257/events.out.tfevents.1736513441.172d0f35d57d.2175.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e2ea3b7c655cbc075f0e1b69bcbcda1211fc6731f37a6d99f56cf68586f6982c
3
+ size 2438
logs/instruct-pix2pix/1736513441.0499015/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: instruction-tuning-sd/cartoonization
14
+ edit_prompt_column: edit_prompt
15
+ edited_image_column: cartoonized_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: 0
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 5000
29
+ mixed_precision: fp16
30
+ non_ema_revision: null
31
+ num_train_epochs: 16
32
+ num_validation_images: 4
33
+ original_image_column: original_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.1736513441.172d0f35d57d.2175.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9514e2bd3df93c94022016018ba5629de246684a58cb0af5a8aaa7c90834f02f
3
+ size 244961
model_index.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "StableDiffusionInstructPix2PixPipeline",
3
- "_diffusers_version": "0.30.0.dev0",
4
  "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
  "feature_extractor": [
6
  "transformers",
 
1
  {
2
  "_class_name": "StableDiffusionInstructPix2PixPipeline",
3
+ "_diffusers_version": "0.33.0.dev0",
4
  "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
  "feature_extractor": [
6
  "transformers",
safety_checker/config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/root/.cache/huggingface/hub/models--runwayml--stable-diffusion-v1-5/snapshots/1d0c4ebf6ff58a5caecab40fa1406526bca4b5b9/safety_checker",
3
  "architectures": [
4
  "StableDiffusionSafetyChecker"
5
  ],
@@ -15,8 +15,9 @@
15
  "num_attention_heads": 12
16
  },
17
  "torch_dtype": "float32",
18
- "transformers_version": "4.41.2",
19
  "vision_config": {
 
20
  "dropout": 0.0,
21
  "hidden_size": 1024,
22
  "intermediate_size": 4096,
 
1
  {
2
+ "_name_or_path": "/root/.cache/huggingface/hub/models--runwayml--stable-diffusion-v1-5/snapshots/451f4fe16113bff5a5d2269ed5ad43b0592e9a14/safety_checker",
3
  "architectures": [
4
  "StableDiffusionSafetyChecker"
5
  ],
 
15
  "num_attention_heads": 12
16
  },
17
  "torch_dtype": "float32",
18
+ "transformers_version": "4.47.1",
19
  "vision_config": {
20
+ "_attn_implementation_autoset": true,
21
  "dropout": 0.0,
22
  "hidden_size": 1024,
23
  "intermediate_size": 4096,
scheduler/scheduler_config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "PNDMScheduler",
3
- "_diffusers_version": "0.30.0.dev0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
 
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,
text_encoder/config.json CHANGED
@@ -20,6 +20,6 @@
20
  "pad_token_id": 1,
21
  "projection_dim": 768,
22
  "torch_dtype": "float16",
23
- "transformers_version": "4.41.2",
24
  "vocab_size": 49408
25
  }
 
20
  "pad_token_id": 1,
21
  "projection_dim": 768,
22
  "torch_dtype": "float16",
23
+ "transformers_version": "4.47.1",
24
  "vocab_size": 49408
25
  }
tokenizer/tokenizer_config.json CHANGED
@@ -19,10 +19,11 @@
19
  }
20
  },
21
  "bos_token": "<|startoftext|>",
22
- "clean_up_tokenization_spaces": true,
23
  "do_lower_case": true,
24
  "eos_token": "<|endoftext|>",
25
  "errors": "replace",
 
26
  "model_max_length": 77,
27
  "pad_token": "<|endoftext|>",
28
  "tokenizer_class": "CLIPTokenizer",
 
19
  }
20
  },
21
  "bos_token": "<|startoftext|>",
22
+ "clean_up_tokenization_spaces": false,
23
  "do_lower_case": true,
24
  "eos_token": "<|endoftext|>",
25
  "errors": "replace",
26
+ "extra_special_tokens": {},
27
  "model_max_length": 77,
28
  "pad_token": "<|endoftext|>",
29
  "tokenizer_class": "CLIPTokenizer",
unet/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
- "_diffusers_version": "0.30.0.dev0",
4
  "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.33.0.dev0",
4
  "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fbabcc9cf08b34fc51fbb198755f700f45e89ed92dcfbb82545f1ff6fa5bc656
3
  size 3438213624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ccf3f72480160ea25661bfd9c48ed32352e9da426686b695207bb873d3d8c07a
3
  size 3438213624
vae/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
- "_diffusers_version": "0.30.0.dev0",
4
  "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
  "act_fn": "silu",
6
  "block_out_channels": [
@@ -21,6 +21,7 @@
21
  "latents_mean": null,
22
  "latents_std": null,
23
  "layers_per_block": 2,
 
24
  "norm_num_groups": 32,
25
  "out_channels": 3,
26
  "sample_size": 512,
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
+ "_diffusers_version": "0.33.0.dev0",
4
  "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
  "act_fn": "silu",
6
  "block_out_channels": [
 
21
  "latents_mean": null,
22
  "latents_std": null,
23
  "layers_per_block": 2,
24
+ "mid_block_add_attention": true,
25
  "norm_num_groups": 32,
26
  "out_channels": 3,
27
  "sample_size": 512,