krhari commited on
Commit
7bef091
·
verified ·
1 Parent(s): 61a6431

End of training

Browse files
README.md CHANGED
@@ -10,6 +10,11 @@ tags:
10
  - diffusers-training
11
  - stable-diffusion
12
  - stable-diffusion-diffusers
 
 
 
 
 
13
  ---
14
 
15
  <!-- This model card has been generated automatically according to the information the training script had access to. You
 
10
  - diffusers-training
11
  - stable-diffusion
12
  - stable-diffusion-diffusers
13
+ - text-to-image
14
+ - dreambooth
15
+ - diffusers-training
16
+ - stable-diffusion
17
+ - stable-diffusion-diffusers
18
  ---
19
 
20
  <!-- This model card has been generated automatically according to the information the training script had access to. You
checkpoint-500/optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bfca0c57cfb614084b83aef44283b772da5b8831d3b31355d76f6727dfd10c74
3
  size 1748609210
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:332ac2757bdc17226b28a614c7809d6bf3f91e9b2cb84a707cf8b62fcf98885e
3
  size 1748609210
checkpoint-500/random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6b7c3b66bb10715cd8ac799fee3ad2eb460ada6ce4810dd7a40a425e051f5940
3
  size 14344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6fecc8817cc206185f1cccf2d0dfeea0bd2338ad21ade5f9294438b67ab906a8
3
  size 14344
checkpoint-500/unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f4705cf923a5662e61d46b45616e5223b3005c1de85116da9894bdcd15cb20a8
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5bda3a219c7d8048b8ca0d3fdcc584ff0efb766b36f8e3fef34df1e4af84615
3
  size 3438167536
logs/dreambooth/1733110075.347154/events.out.tfevents.1733110075.f94ca8423315.9728.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dcd10ce3ff7e769dc4307ea7f96bc4fd484d7c7a1c77b9eb50b4afb2606f1886
3
+ size 2936
logs/dreambooth/1733110075.352346/hparams.yml ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: /content/diffusers/examples/dreambooth/dog
10
+ class_labels_conditioning: null
11
+ class_prompt: a photo of dog
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: /content/diffusers/examples/dreambooth/dog
19
+ instance_prompt: my dog teddy
20
+ learning_rate: 2.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: 5
31
+ num_train_epochs: 115
32
+ num_validation_images: 4
33
+ offset_noise: false
34
+ output_dir: /content/models/dreambooth_output
35
+ pre_compute_text_embeddings: false
36
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
37
+ prior_generation_precision: null
38
+ prior_loss_weight: 1.0
39
+ push_to_hub: true
40
+ report_to: tensorboard
41
+ resolution: 512
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: true
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
+ variant: null
60
+ with_prior_preservation: true
logs/dreambooth/1733110129.1800492/events.out.tfevents.1733110129.f94ca8423315.10094.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23bf972baadcfd163c78d3839f350eca501a2307b6ffdb3414b3029c8d950abe
3
+ size 2936
logs/dreambooth/1733110129.181825/hparams.yml ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: /content/diffusers/examples/dreambooth/dog
10
+ class_labels_conditioning: null
11
+ class_prompt: a photo of dog
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: /content/diffusers/examples/dreambooth/dog
19
+ instance_prompt: my dog teddy
20
+ learning_rate: 2.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: 5
31
+ num_train_epochs: 160
32
+ num_validation_images: 4
33
+ offset_noise: false
34
+ output_dir: /content/models/dreambooth_output
35
+ pre_compute_text_embeddings: false
36
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
37
+ prior_generation_precision: null
38
+ prior_loss_weight: 1.0
39
+ push_to_hub: true
40
+ report_to: tensorboard
41
+ resolution: 512
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: true
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
+ variant: null
60
+ with_prior_preservation: true
logs/dreambooth/events.out.tfevents.1733110075.f94ca8423315.9728.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:baa4bee6847324ecbf3c945955b622fbda069be3d71c50ec26bc9a460c22c21f
3
+ size 252
logs/dreambooth/events.out.tfevents.1733110129.f94ca8423315.10094.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:502de724cd37681508b5e323d88da13f9605339dcaddd8c39208533a8ec4058b
3
+ size 67034
safety_checker/config.json CHANGED
@@ -15,7 +15,7 @@
15
  "num_attention_heads": 12
16
  },
17
  "torch_dtype": "float32",
18
- "transformers_version": "4.46.3",
19
  "vision_config": {
20
  "_attn_implementation_autoset": true,
21
  "dropout": 0.0,
 
15
  "num_attention_heads": 12
16
  },
17
  "torch_dtype": "float32",
18
+ "transformers_version": "4.46.2",
19
  "vision_config": {
20
  "_attn_implementation_autoset": true,
21
  "dropout": 0.0,
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.46.3",
24
  "vocab_size": 49408
25
  }
 
20
  "pad_token_id": 1,
21
  "projection_dim": 512,
22
  "torch_dtype": "float32",
23
+ "transformers_version": "4.46.2",
24
  "vocab_size": 49408
25
  }
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e364d69e7cf4dfa515897dbf70c25e7c6ad43bd9ee2173a4955c329918cacf5f
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5310616b64ad7d0eb9d52066f5b79d4eceed374dbf3e745371e5bc2d971d2ed5
3
  size 3438167536