askulkarni2 commited on
Commit
6f381d3
·
1 Parent(s): dba4a1b

End of training

Browse files
checkpoint-500/optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1226c9bfe28b7773bc86e3a6eb8867ee50372afd3f82ddbb5bbc1f611d164d6d
3
  size 1737915525
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9b849df631c0583987ec939fdfe3cd5b42de498cb3bf0ac030038f7fe793b6d
3
  size 1737915525
checkpoint-500/random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5be865aa21e661e2c27eb735b5d365ee5b6090ce671642f04fba5ac4cf9fa67d
3
  size 14663
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7305e8362e9a273f741104de310f26d7ee656bc24ec8ce3d4e0c4e807ca8d731
3
  size 14663
checkpoint-500/unet/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
- "_diffusers_version": "0.21.0.dev0",
4
  "_name_or_path": "stabilityai/stable-diffusion-2-1",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
@@ -33,6 +33,7 @@
33
  "DownBlock2D"
34
  ],
35
  "downsample_padding": 1,
 
36
  "dual_cross_attention": false,
37
  "encoder_hid_dim": null,
38
  "encoder_hid_dim_type": null,
@@ -53,6 +54,7 @@
53
  "resnet_out_scale_factor": 1.0,
54
  "resnet_skip_time_act": false,
55
  "resnet_time_scale_shift": "default",
 
56
  "sample_size": 96,
57
  "time_cond_proj_dim": null,
58
  "time_embedding_act_fn": null,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.24.0.dev0",
4
  "_name_or_path": "stabilityai/stable-diffusion-2-1",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
 
33
  "DownBlock2D"
34
  ],
35
  "downsample_padding": 1,
36
+ "dropout": 0.0,
37
  "dual_cross_attention": false,
38
  "encoder_hid_dim": null,
39
  "encoder_hid_dim_type": null,
 
54
  "resnet_out_scale_factor": 1.0,
55
  "resnet_skip_time_act": false,
56
  "resnet_time_scale_shift": "default",
57
+ "reverse_transformer_layers_per_block": null,
58
  "sample_size": 96,
59
  "time_cond_proj_dim": null,
60
  "time_embedding_act_fn": null,
checkpoint-500/unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b36cfd89655bd3efd9715f4513a0a215626fd77c0b7c981ab64c4764ff2fd694
3
  size 3463726504
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:62e659ba49844fd29fb47fe289602b3ef5dae59d8d974a45decb3156e444c689
3
  size 3463726504
feature_extractor/preprocessor_config.json CHANGED
@@ -24,5 +24,6 @@
24
  "rescale_factor": 0.00392156862745098,
25
  "size": {
26
  "shortest_edge": 224
27
- }
 
28
  }
 
24
  "rescale_factor": 0.00392156862745098,
25
  "size": {
26
  "shortest_edge": 224
27
+ },
28
+ "use_square_size": false
29
  }
logs/dreambooth/1699952350.258148/events.out.tfevents.1699952350.jupyter-user-2d1.221.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99527008254f84906318c022f9d9c10ca46b5e53eb1c98c04f7803b6bcee7f9c
3
+ size 2696
logs/dreambooth/1699952350.2594075/hparams.yml ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ center_crop: false
7
+ checkpointing_steps: 500
8
+ checkpoints_total_limit: null
9
+ class_data_dir: null
10
+ class_labels_conditioning: null
11
+ class_prompt: null
12
+ dataloader_num_workers: 0
13
+ enable_xformers_memory_efficient_attention: true
14
+ gradient_accumulation_steps: 1
15
+ gradient_checkpointing: true
16
+ hub_model_id: null
17
+ hub_token: null
18
+ instance_data_dir: dog
19
+ instance_prompt: a photo of [v]dog
20
+ learning_rate: 1.0e-06
21
+ local_rank: -1
22
+ logging_dir: logs
23
+ lr_num_cycles: 1
24
+ lr_power: 1.0
25
+ lr_scheduler: constant
26
+ lr_warmup_steps: 0
27
+ max_grad_norm: 1.0
28
+ max_train_steps: 800
29
+ mixed_precision: null
30
+ num_class_images: 100
31
+ num_train_epochs: 160
32
+ num_validation_images: 4
33
+ offset_noise: false
34
+ output_dir: dogbooth
35
+ pre_compute_text_embeddings: false
36
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1
37
+ prior_generation_precision: null
38
+ prior_loss_weight: 1.0
39
+ push_to_hub: true
40
+ report_to: tensorboard
41
+ resolution: 768
42
+ resume_from_checkpoint: null
43
+ revision: null
44
+ sample_batch_size: 4
45
+ scale_lr: false
46
+ seed: null
47
+ set_grads_to_none: false
48
+ skip_save_text_encoder: false
49
+ snr_gamma: null
50
+ text_encoder_use_attention_mask: false
51
+ tokenizer_max_length: null
52
+ tokenizer_name: null
53
+ train_batch_size: 1
54
+ train_text_encoder: false
55
+ use_8bit_adam: true
56
+ validation_prompt: null
57
+ validation_scheduler: DPMSolverMultistepScheduler
58
+ validation_steps: 100
59
+ with_prior_preservation: false
logs/dreambooth/events.out.tfevents.1699952350.jupyter-user-2d1.221.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31848ffbdfb8ee513675a2bd6412a299de343828e9600f59ba5e076f5b59b2d9
3
+ size 66986
model_index.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
- "_diffusers_version": "0.22.0.dev0",
4
  "_name_or_path": "stabilityai/stable-diffusion-2-1",
5
  "feature_extractor": [
6
  "transformers",
 
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
+ "_diffusers_version": "0.24.0.dev0",
4
  "_name_or_path": "stabilityai/stable-diffusion-2-1",
5
  "feature_extractor": [
6
  "transformers",
scheduler/scheduler_config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "DDIMScheduler",
3
- "_diffusers_version": "0.22.0.dev0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
 
1
  {
2
  "_class_name": "DDIMScheduler",
3
+ "_diffusers_version": "0.24.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": 512,
22
  "torch_dtype": "float32",
23
- "transformers_version": "4.34.1",
24
  "vocab_size": 49408
25
  }
 
20
  "pad_token_id": 1,
21
  "projection_dim": 512,
22
  "torch_dtype": "float32",
23
+ "transformers_version": "4.35.0",
24
  "vocab_size": 49408
25
  }
unet/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
- "_diffusers_version": "0.22.0.dev0",
4
  "_name_or_path": "stabilityai/stable-diffusion-2-1",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.24.0.dev0",
4
  "_name_or_path": "stabilityai/stable-diffusion-2-1",
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:f57d6d2857726ccf92833552c10970c367e33eafdac4527ddb39da740af88800
3
  size 3463726504
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e1a671c6cfdca676388c8f7a85ddcf9ff506a6251835196281c2f75b7b86014c
3
  size 3463726504
vae/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
- "_diffusers_version": "0.22.0.dev0",
4
  "_name_or_path": "/home/jovyan/.cache/huggingface/hub/models--stabilityai--stable-diffusion-2-1/snapshots/5cae40e6a2745ae2b01ad92ae5043f95f23644d6/vae",
5
  "act_fn": "silu",
6
  "block_out_channels": [
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
+ "_diffusers_version": "0.24.0.dev0",
4
  "_name_or_path": "/home/jovyan/.cache/huggingface/hub/models--stabilityai--stable-diffusion-2-1/snapshots/5cae40e6a2745ae2b01ad92ae5043f95f23644d6/vae",
5
  "act_fn": "silu",
6
  "block_out_channels": [