Hugging Face's logo

AvinashAmballa
/
custom_cond

Text-to-Image
Diffusers
TensorBoard
Safetensors
StableDiffusionPipeline
dreambooth
diffusers-training
stable-diffusion
stable-diffusion-diffusers
Model card Files Metrics Community
AvinashAmballa commited on
Commit
3d7d298
·
verified ·
1 Parent(s): c991dbd

End of training

Browse files
Files changed (16) hide show
  1. logs/dreambooth/1737567214.8243651/events.out.tfevents.1737567214.gpu012.386624.1 +3 -0
  2. logs/dreambooth/1737567214.8411522/hparams.yml +61 -0
  3. logs/dreambooth/1737573334.9626033/events.out.tfevents.1737573334.gpu012.399670.1 +3 -0
  4. logs/dreambooth/1737573334.9808347/hparams.yml +61 -0
  5. logs/dreambooth/1737573540.3285098/events.out.tfevents.1737573540.gpu012.400464.1 +3 -0
  6. logs/dreambooth/1737573540.3637197/hparams.yml +61 -0
  7. logs/dreambooth/1737573861.8629513/events.out.tfevents.1737573861.gpu012.401768.1 +3 -0
  8. logs/dreambooth/1737573862.0484698/hparams.yml +61 -0
  9. logs/dreambooth/1737574170.8181684/events.out.tfevents.1737574170.gpu012.402854.1 +3 -0
  10. logs/dreambooth/1737574170.8476746/hparams.yml +61 -0
  11. logs/dreambooth/events.out.tfevents.1737567214.gpu012.386624.0 +3 -0
  12. logs/dreambooth/events.out.tfevents.1737573334.gpu012.399670.0 +3 -0
  13. logs/dreambooth/events.out.tfevents.1737573540.gpu012.400464.0 +3 -0
  14. logs/dreambooth/events.out.tfevents.1737573861.gpu012.401768.0 +3 -0
  15. logs/dreambooth/events.out.tfevents.1737574170.gpu012.402854.0 +3 -0
  16. unet/diffusion_pytorch_model.safetensors +1 -1
logs/dreambooth/1737567214.8243651/events.out.tfevents.1737567214.gpu012.386624.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb73bc294db9798560aca903f13902a984fe4d9e5b8dbcd55d1ed8aeee06df63
3
+ size 3018
logs/dreambooth/1737567214.8411522/hparams.yml ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: VISA/custom/normal
10
+ class_labels_conditioning: null
11
+ class_prompt: a photo of candle, a photo of capsule, a photo of cashew
12
+ dataloader_num_workers: 0
13
+ enable_xformers_memory_efficient_attention: false
14
+ gradient_accumulation_steps: 1
15
+ gradient_checkpointing: false
16
+ hub_model_id: null
17
+ hub_token: null
18
+ instance_data_dir: VISA/custom/anomaly
19
+ instance_prompt: a photo of candle with dru anomaly, a photo of capsule with ibc anomaly,
20
+ a photo of cashew with ucn anomaly
21
+ learning_rate: 5.0e-06
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_num_cycles: 1
25
+ lr_power: 1.0
26
+ lr_scheduler: constant
27
+ lr_warmup_steps: 0
28
+ max_grad_norm: 1.0
29
+ max_train_steps: 400
30
+ mixed_precision: null
31
+ num_class_images: 18
32
+ num_train_epochs: 23
33
+ num_validation_images: 4
34
+ offset_noise: false
35
+ output_dir: results/VISA/custom_cond
36
+ pre_compute_text_embeddings: false
37
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
38
+ prior_generation_precision: null
39
+ prior_loss_weight: 1.0
40
+ push_to_hub: true
41
+ report_to: tensorboard
42
+ resolution: 512
43
+ resume_from_checkpoint: null
44
+ revision: null
45
+ sample_batch_size: 4
46
+ scale_lr: false
47
+ seed: null
48
+ set_grads_to_none: false
49
+ skip_save_text_encoder: false
50
+ snr_gamma: null
51
+ text_encoder_use_attention_mask: false
52
+ tokenizer_max_length: null
53
+ tokenizer_name: null
54
+ train_batch_size: 1
55
+ train_text_encoder: false
56
+ use_8bit_adam: false
57
+ validation_prompt: null
58
+ validation_scheduler: DPMSolverMultistepScheduler
59
+ validation_steps: 100
60
+ variant: null
61
+ with_prior_preservation: true
logs/dreambooth/1737573334.9626033/events.out.tfevents.1737573334.gpu012.399670.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67ffdabeb94c4b05f55bf7294da7fb11e851c8e4404d792af1a77e55150f185e
3
+ size 3018
logs/dreambooth/1737573334.9808347/hparams.yml ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: VISA/custom/normal
10
+ class_labels_conditioning: null
11
+ class_prompt: a photo of candle, a photo of capsule, a photo of cashew
12
+ dataloader_num_workers: 0
13
+ enable_xformers_memory_efficient_attention: false
14
+ gradient_accumulation_steps: 1
15
+ gradient_checkpointing: false
16
+ hub_model_id: null
17
+ hub_token: null
18
+ instance_data_dir: VISA/custom/anomaly
19
+ instance_prompt: a photo of candle with dru anomaly, a photo of capsule with ibc anomaly,
20
+ a photo of cashew with ucn anomaly
21
+ learning_rate: 5.0e-06
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_num_cycles: 1
25
+ lr_power: 1.0
26
+ lr_scheduler: constant
27
+ lr_warmup_steps: 0
28
+ max_grad_norm: 1.0
29
+ max_train_steps: 400
30
+ mixed_precision: null
31
+ num_class_images: 18
32
+ num_train_epochs: 23
33
+ num_validation_images: 4
34
+ offset_noise: false
35
+ output_dir: results/VISA/custom_cond
36
+ pre_compute_text_embeddings: false
37
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
38
+ prior_generation_precision: null
39
+ prior_loss_weight: 1.0
40
+ push_to_hub: true
41
+ report_to: tensorboard
42
+ resolution: 512
43
+ resume_from_checkpoint: null
44
+ revision: null
45
+ sample_batch_size: 4
46
+ scale_lr: false
47
+ seed: null
48
+ set_grads_to_none: false
49
+ skip_save_text_encoder: false
50
+ snr_gamma: null
51
+ text_encoder_use_attention_mask: false
52
+ tokenizer_max_length: null
53
+ tokenizer_name: null
54
+ train_batch_size: 1
55
+ train_text_encoder: false
56
+ use_8bit_adam: false
57
+ validation_prompt: null
58
+ validation_scheduler: DPMSolverMultistepScheduler
59
+ validation_steps: 100
60
+ variant: null
61
+ with_prior_preservation: true
logs/dreambooth/1737573540.3285098/events.out.tfevents.1737573540.gpu012.400464.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f9a92777916213a2bd697ecbb8769a21ebd8f7d001ffd54d2364d827ba9ecb9
3
+ size 3018
logs/dreambooth/1737573540.3637197/hparams.yml ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: VISA/custom/normal
10
+ class_labels_conditioning: null
11
+ class_prompt: a photo of candle, a photo of capsule, a photo of cashew
12
+ dataloader_num_workers: 0
13
+ enable_xformers_memory_efficient_attention: false
14
+ gradient_accumulation_steps: 1
15
+ gradient_checkpointing: false
16
+ hub_model_id: null
17
+ hub_token: null
18
+ instance_data_dir: VISA/custom/anomaly
19
+ instance_prompt: a photo of candle with dru anomaly, a photo of capsule with ibc anomaly,
20
+ a photo of cashew with ucn anomaly
21
+ learning_rate: 5.0e-06
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_num_cycles: 1
25
+ lr_power: 1.0
26
+ lr_scheduler: constant
27
+ lr_warmup_steps: 0
28
+ max_grad_norm: 1.0
29
+ max_train_steps: 400
30
+ mixed_precision: null
31
+ num_class_images: 18
32
+ num_train_epochs: 23
33
+ num_validation_images: 4
34
+ offset_noise: false
35
+ output_dir: results/VISA/custom_cond
36
+ pre_compute_text_embeddings: false
37
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
38
+ prior_generation_precision: null
39
+ prior_loss_weight: 1.0
40
+ push_to_hub: true
41
+ report_to: tensorboard
42
+ resolution: 512
43
+ resume_from_checkpoint: null
44
+ revision: null
45
+ sample_batch_size: 4
46
+ scale_lr: false
47
+ seed: null
48
+ set_grads_to_none: false
49
+ skip_save_text_encoder: false
50
+ snr_gamma: null
51
+ text_encoder_use_attention_mask: false
52
+ tokenizer_max_length: null
53
+ tokenizer_name: null
54
+ train_batch_size: 1
55
+ train_text_encoder: false
56
+ use_8bit_adam: false
57
+ validation_prompt: null
58
+ validation_scheduler: DPMSolverMultistepScheduler
59
+ validation_steps: 100
60
+ variant: null
61
+ with_prior_preservation: true
logs/dreambooth/1737573861.8629513/events.out.tfevents.1737573861.gpu012.401768.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:db99d15eedab463d83e8fbe9f364926cc3af194534f8bb4ccfa02ab56c7506c0
3
+ size 3018
logs/dreambooth/1737573862.0484698/hparams.yml ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: VISA/custom/normal
10
+ class_labels_conditioning: null
11
+ class_prompt: a photo of candle, a photo of capsule, a photo of cashew
12
+ dataloader_num_workers: 0
13
+ enable_xformers_memory_efficient_attention: false
14
+ gradient_accumulation_steps: 1
15
+ gradient_checkpointing: false
16
+ hub_model_id: null
17
+ hub_token: null
18
+ instance_data_dir: VISA/custom/anomaly
19
+ instance_prompt: a photo of candle with dru anomaly, a photo of capsule with ibc anomaly,
20
+ a photo of cashew with ucn anomaly
21
+ learning_rate: 5.0e-06
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_num_cycles: 1
25
+ lr_power: 1.0
26
+ lr_scheduler: constant
27
+ lr_warmup_steps: 0
28
+ max_grad_norm: 1.0
29
+ max_train_steps: 400
30
+ mixed_precision: null
31
+ num_class_images: 18
32
+ num_train_epochs: 23
33
+ num_validation_images: 4
34
+ offset_noise: false
35
+ output_dir: results/VISA/custom_cond
36
+ pre_compute_text_embeddings: false
37
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
38
+ prior_generation_precision: null
39
+ prior_loss_weight: 1.0
40
+ push_to_hub: true
41
+ report_to: tensorboard
42
+ resolution: 512
43
+ resume_from_checkpoint: null
44
+ revision: null
45
+ sample_batch_size: 4
46
+ scale_lr: false
47
+ seed: null
48
+ set_grads_to_none: false
49
+ skip_save_text_encoder: false
50
+ snr_gamma: null
51
+ text_encoder_use_attention_mask: false
52
+ tokenizer_max_length: null
53
+ tokenizer_name: null
54
+ train_batch_size: 1
55
+ train_text_encoder: false
56
+ use_8bit_adam: false
57
+ validation_prompt: null
58
+ validation_scheduler: DPMSolverMultistepScheduler
59
+ validation_steps: 100
60
+ variant: null
61
+ with_prior_preservation: true
logs/dreambooth/1737574170.8181684/events.out.tfevents.1737574170.gpu012.402854.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a2dab9cfbb8e63c6083265454b2cd9add998b1df7f99e048959e73bffa63e6f
3
+ size 3018
logs/dreambooth/1737574170.8476746/hparams.yml ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: VISA/custom/normal
10
+ class_labels_conditioning: null
11
+ class_prompt: a photo of candle, a photo of capsule, a photo of cashew
12
+ dataloader_num_workers: 0
13
+ enable_xformers_memory_efficient_attention: false
14
+ gradient_accumulation_steps: 1
15
+ gradient_checkpointing: false
16
+ hub_model_id: null
17
+ hub_token: null
18
+ instance_data_dir: VISA/custom/anomaly
19
+ instance_prompt: a photo of candle with dru anomaly, a photo of capsule with ibc anomaly,
20
+ a photo of cashew with ucn anomaly
21
+ learning_rate: 5.0e-06
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_num_cycles: 1
25
+ lr_power: 1.0
26
+ lr_scheduler: constant
27
+ lr_warmup_steps: 0
28
+ max_grad_norm: 1.0
29
+ max_train_steps: 400
30
+ mixed_precision: null
31
+ num_class_images: 18
32
+ num_train_epochs: 23
33
+ num_validation_images: 4
34
+ offset_noise: false
35
+ output_dir: results/VISA/custom_cond
36
+ pre_compute_text_embeddings: false
37
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
38
+ prior_generation_precision: null
39
+ prior_loss_weight: 1.0
40
+ push_to_hub: true
41
+ report_to: tensorboard
42
+ resolution: 512
43
+ resume_from_checkpoint: null
44
+ revision: null
45
+ sample_batch_size: 4
46
+ scale_lr: false
47
+ seed: null
48
+ set_grads_to_none: false
49
+ skip_save_text_encoder: false
50
+ snr_gamma: null
51
+ text_encoder_use_attention_mask: false
52
+ tokenizer_max_length: null
53
+ tokenizer_name: null
54
+ train_batch_size: 1
55
+ train_text_encoder: false
56
+ use_8bit_adam: false
57
+ validation_prompt: null
58
+ validation_scheduler: DPMSolverMultistepScheduler
59
+ validation_steps: 100
60
+ variant: null
61
+ with_prior_preservation: true
logs/dreambooth/events.out.tfevents.1737567214.gpu012.386624.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce82cb5935c08062c34c6ca4e013246f9729bc14caff8d67827506b8e2c87d6c
3
+ size 4844
logs/dreambooth/events.out.tfevents.1737573334.gpu012.399670.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ac76e85a0b28846a18ffcb4ea729a28047ae2d63561eddbf2c3e38663d4933d
3
+ size 88
logs/dreambooth/events.out.tfevents.1737573540.gpu012.400464.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ab987c07aec1c911038e220a325ab4b04653bf3ff416efcfa344e3d65735061
3
+ size 908
logs/dreambooth/events.out.tfevents.1737573861.gpu012.401768.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a5c0c869138d54161f95f9efa7799a05bd426730a5fd81a91a56e0a8505e8b6
3
+ size 88
logs/dreambooth/events.out.tfevents.1737574170.gpu012.402854.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37299a8985918c14d4c86389ffa2448e6c9e23ca5d615fcf37530834824b2105
3
+ size 33434
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8c2dbdedf510be42eb13e429ff1f91d465b787e1c8b88dff968d82a86a9bf274
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b0c3faadb24737de19f097eb3b880bde4a040961d4a9d8d95f0cd92efea86ffa
3
  size 3438167536