Hugging Face's logo

AvinashAmballa
/
custom_cond

Text-to-Image
Diffusers
TensorBoard
Safetensors
StableDiffusionPipeline
dreambooth
diffusers-training
stable-diffusion
stable-diffusion-diffusers
Inference Endpoints
Model card Files Metrics Community
AvinashAmballa commited on
Commit
97b636a
·
verified ·
1 Parent(s): c4fde2d

End of training

Browse files
Files changed (16) hide show
  1. logs/dreambooth/1737611223.383769/events.out.tfevents.1737611223.gypsum-gpu005.1382239.1 +3 -0
  2. logs/dreambooth/1737611223.3965268/hparams.yml +61 -0
  3. logs/dreambooth/1737649955.46876/events.out.tfevents.1737649955.gypsum-gpu003.1719243.1 +3 -0
  4. logs/dreambooth/1737649955.4845476/hparams.yml +61 -0
  5. logs/dreambooth/1737651811.329677/events.out.tfevents.1737651811.gypsum-gpu003.1721994.1 +3 -0
  6. logs/dreambooth/1737651811.3791997/hparams.yml +61 -0
  7. logs/dreambooth/1737651955.8414671/events.out.tfevents.1737651955.gypsum-gpu003.1722041.1 +3 -0
  8. logs/dreambooth/1737651955.8617065/hparams.yml +61 -0
  9. logs/dreambooth/1737695825.4787703/events.out.tfevents.1737695825.gpu011.261867.1 +3 -0
  10. logs/dreambooth/1737695825.5344892/hparams.yml +62 -0
  11. logs/dreambooth/events.out.tfevents.1737611223.gypsum-gpu005.1382239.0 +3 -0
  12. logs/dreambooth/events.out.tfevents.1737649955.gypsum-gpu003.1719243.0 +3 -0
  13. logs/dreambooth/events.out.tfevents.1737651811.gypsum-gpu003.1721994.0 +3 -0
  14. logs/dreambooth/events.out.tfevents.1737651955.gypsum-gpu003.1722041.0 +3 -0
  15. logs/dreambooth/events.out.tfevents.1737695825.gpu011.261867.0 +3 -0
  16. unet/diffusion_pytorch_model.safetensors +1 -1
logs/dreambooth/1737611223.383769/events.out.tfevents.1737611223.gypsum-gpu005.1382239.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f55d8d930302c65eb2db8e5967d824ff9ff60114235e4f388c00d091fe1fc239
3
+ size 3018
logs/dreambooth/1737611223.3965268/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/1737649955.46876/events.out.tfevents.1737649955.gypsum-gpu003.1719243.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf4d68f0d15750ff54422be45564793b344ec7393950ac2847b48dc7516893d7
3
+ size 3018
logs/dreambooth/1737649955.4845476/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/1737651811.329677/events.out.tfevents.1737651811.gypsum-gpu003.1721994.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c5eb368a1db41af10c243de897d1fe3103ee12427e842b8591c0c40214571841
3
+ size 3018
logs/dreambooth/1737651811.3791997/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/1737651955.8414671/events.out.tfevents.1737651955.gypsum-gpu003.1722041.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9010bf392036fe5b9413f5287a1520fa593d9096e5d3dc8220bcca5055ae7b49
3
+ size 3018
logs/dreambooth/1737651955.8617065/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/1737695825.4787703/events.out.tfevents.1737695825.gpu011.261867.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3887fd391ad2f8646c2c3a5ecf91aedf56cdaccb15d24a01ad5d7aba25762a4c
3
+ size 3083
logs/dreambooth/1737695825.5344892/hparams.yml ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 dru candle, a photo of ibc capsule, a photo of ucn cashew
20
+ learning_rate: 5.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
+ mask_data_dir: VISA/custom/mask
28
+ mask_prompt: dru, ibc, ucn
29
+ max_grad_norm: 1.0
30
+ max_train_steps: 400
31
+ mixed_precision: null
32
+ num_class_images: 18
33
+ num_train_epochs: 23
34
+ num_validation_images: 4
35
+ offset_noise: false
36
+ output_dir: results/VISA/custom_cond
37
+ pre_compute_text_embeddings: false
38
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
39
+ prior_generation_precision: null
40
+ prior_loss_weight: 1.0
41
+ push_to_hub: true
42
+ report_to: tensorboard
43
+ resolution: 512
44
+ resume_from_checkpoint: null
45
+ revision: null
46
+ sample_batch_size: 4
47
+ scale_lr: false
48
+ seed: null
49
+ set_grads_to_none: false
50
+ skip_save_text_encoder: false
51
+ snr_gamma: null
52
+ text_encoder_use_attention_mask: false
53
+ tokenizer_max_length: null
54
+ tokenizer_name: null
55
+ train_batch_size: 1
56
+ train_text_encoder: false
57
+ use_8bit_adam: false
58
+ validation_prompt: null
59
+ validation_scheduler: DPMSolverMultistepScheduler
60
+ validation_steps: 100
61
+ variant: null
62
+ with_prior_preservation: true
logs/dreambooth/events.out.tfevents.1737611223.gypsum-gpu005.1382239.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:52c6263852fb80a0f707f21d5e98519af71655e6f74a27638fd31154dfaee9cb
3
+ size 252
logs/dreambooth/events.out.tfevents.1737649955.gypsum-gpu003.1719243.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54194a26e60418adf8f9c0796311c61ffdabab52e151461553782d7da1ad5a47
3
+ size 88
logs/dreambooth/events.out.tfevents.1737651811.gypsum-gpu003.1721994.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b283b87d78c56cfbb7fb0e294aad5bcbac18d2e9b06595d0b742256362fb2245
3
+ size 88
logs/dreambooth/events.out.tfevents.1737651955.gypsum-gpu003.1722041.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ab1e9a5175016cfd4d440cc9bec4bf8b936c0a7d3bff7e40d394be0b687918d
3
+ size 88
logs/dreambooth/events.out.tfevents.1737695825.gpu011.261867.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:07bbcba074049d47398ccb83180e3db534a4b6714ef6d4401e56782f21cd19e0
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:9041485d0393ffa77995693d3ac0d2f80cc592a56e24a53f92a8ae344bea674c
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:85f3166075bad1791d0c6e566f08ac61477eabe5fba989fda485b8f49ee16017
3
  size 3438167536