Hugging Face's logo

Jise
/
controlnet-X-ray

Text-to-Image
Diffusers
TensorBoard
stable-diffusion
stable-diffusion-diffusers
controlnet
Model card Files Metrics Community
1
Jise commited on
Commit
493e63e
•
1 Parent(s): f985d76

End of training

Browse files
Files changed (41) hide show
  1. README.md +20 -0
  2. checkpoint-1000/controlnet/config.json +44 -0
  3. checkpoint-1000/controlnet/diffusion_pytorch_model.bin +3 -0
  4. checkpoint-1000/optimizer.bin +3 -0
  5. checkpoint-1000/random_states_0.pkl +3 -0
  6. checkpoint-1000/scaler.pt +3 -0
  7. checkpoint-1000/scheduler.bin +3 -0
  8. checkpoint-1500/controlnet/config.json +44 -0
  9. checkpoint-1500/controlnet/diffusion_pytorch_model.bin +3 -0
  10. checkpoint-1500/optimizer.bin +3 -0
  11. checkpoint-1500/random_states_0.pkl +3 -0
  12. checkpoint-1500/scaler.pt +3 -0
  13. checkpoint-1500/scheduler.bin +3 -0
  14. checkpoint-2000/controlnet/config.json +44 -0
  15. checkpoint-2000/controlnet/diffusion_pytorch_model.bin +3 -0
  16. checkpoint-2000/optimizer.bin +3 -0
  17. checkpoint-2000/random_states_0.pkl +3 -0
  18. checkpoint-2000/scaler.pt +3 -0
  19. checkpoint-2000/scheduler.bin +3 -0
  20. checkpoint-500/controlnet/config.json +44 -0
  21. checkpoint-500/controlnet/diffusion_pytorch_model.bin +3 -0
  22. checkpoint-500/optimizer.bin +3 -0
  23. checkpoint-500/random_states_0.pkl +3 -0
  24. checkpoint-500/scaler.pt +3 -0
  25. checkpoint-500/scheduler.bin +3 -0
  26. config.json +44 -0
  27. diffusion_pytorch_model.bin +3 -0
  28. image_control.png +0 -0
  29. images_0.png +0 -0
  30. logs/train_controlnet/1698136955.89282/events.out.tfevents.1698136955.285d4da4d3dd.1278.1 +3 -0
  31. logs/train_controlnet/1698136955.8950262/hparams.yml +49 -0
  32. logs/train_controlnet/1698137667.7316034/events.out.tfevents.1698137667.285d4da4d3dd.5335.1 +3 -0
  33. logs/train_controlnet/1698137667.7340715/hparams.yml +49 -0
  34. logs/train_controlnet/1698138332.2972698/events.out.tfevents.1698138332.285d4da4d3dd.8317.1 +3 -0
  35. logs/train_controlnet/1698138332.2989447/hparams.yml +49 -0
  36. logs/train_controlnet/1698139447.0913413/events.out.tfevents.1698139447.285d4da4d3dd.13142.1 +3 -0
  37. logs/train_controlnet/1698139447.094729/hparams.yml +49 -0
  38. logs/train_controlnet/events.out.tfevents.1698136955.285d4da4d3dd.1278.0 +3 -0
  39. logs/train_controlnet/events.out.tfevents.1698137667.285d4da4d3dd.5335.0 +3 -0
  40. logs/train_controlnet/events.out.tfevents.1698138332.285d4da4d3dd.8317.0 +3 -0
  41. logs/train_controlnet/events.out.tfevents.1698139447.285d4da4d3dd.13142.0 +3 -0
README.md ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ license: creativeml-openrail-m
4
+ base_model: runwayml/stable-diffusion-v1-5
5
+ tags:
6
+ - stable-diffusion
7
+ - stable-diffusion-diffusers
8
+ - text-to-image
9
+ - diffusers
10
+ - controlnet
11
+ inference: true
12
+ ---
13
+
14
+ # controlnet-Jise/controlnet-X-ray
15
+
16
+ These are controlnet weights trained on runwayml/stable-diffusion-v1-5 with new type of conditioning.
17
+ You can find some example images below.
18
+ prompt: chest X_ray
19
+ ![images_0)](./images_0.png)
20
+
checkpoint-1000/controlnet/config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "ControlNetModel",
3
+ "_diffusers_version": "0.18.2",
4
+ "act_fn": "silu",
5
+ "attention_head_dim": 8,
6
+ "block_out_channels": [
7
+ 320,
8
+ 640,
9
+ 1280,
10
+ 1280
11
+ ],
12
+ "class_embed_type": null,
13
+ "conditioning_channels": 3,
14
+ "conditioning_embedding_out_channels": [
15
+ 16,
16
+ 32,
17
+ 96,
18
+ 256
19
+ ],
20
+ "controlnet_conditioning_channel_order": "rgb",
21
+ "cross_attention_dim": 768,
22
+ "down_block_types": [
23
+ "CrossAttnDownBlock2D",
24
+ "CrossAttnDownBlock2D",
25
+ "CrossAttnDownBlock2D",
26
+ "DownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "flip_sin_to_cos": true,
30
+ "freq_shift": 0,
31
+ "global_pool_conditions": false,
32
+ "in_channels": 4,
33
+ "layers_per_block": 2,
34
+ "mid_block_scale_factor": 1,
35
+ "norm_eps": 1e-05,
36
+ "norm_num_groups": 32,
37
+ "num_attention_heads": null,
38
+ "num_class_embeds": null,
39
+ "only_cross_attention": false,
40
+ "projection_class_embeddings_input_dim": null,
41
+ "resnet_time_scale_shift": "default",
42
+ "upcast_attention": false,
43
+ "use_linear_projection": false
44
+ }
checkpoint-1000/controlnet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a31e3846a88febd49e88147b7bea9f55c27c7723a0fb0d6568b2e1e0ea453155
3
+ size 1445259705
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf4747776c0799b9787e09c4a289fb7473ae92f030e8fcac79cab5326126087e
3
+ size 725253381
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b0ccea90a2105bf65ed2616f874024418d9bc5b1259a160807a7fa90b5e0cf59
3
+ size 14663
checkpoint-1000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68cff80b680ddf6e7abbef98b5f336b97f9b5963e2209307f639383870e8cc71
3
+ size 557
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f34912fa092355c5a950e4967ea28cbb3556238a7857baa5999d228cde0ae4c3
3
+ size 563
checkpoint-1500/controlnet/config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "ControlNetModel",
3
+ "_diffusers_version": "0.18.2",
4
+ "act_fn": "silu",
5
+ "attention_head_dim": 8,
6
+ "block_out_channels": [
7
+ 320,
8
+ 640,
9
+ 1280,
10
+ 1280
11
+ ],
12
+ "class_embed_type": null,
13
+ "conditioning_channels": 3,
14
+ "conditioning_embedding_out_channels": [
15
+ 16,
16
+ 32,
17
+ 96,
18
+ 256
19
+ ],
20
+ "controlnet_conditioning_channel_order": "rgb",
21
+ "cross_attention_dim": 768,
22
+ "down_block_types": [
23
+ "CrossAttnDownBlock2D",
24
+ "CrossAttnDownBlock2D",
25
+ "CrossAttnDownBlock2D",
26
+ "DownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "flip_sin_to_cos": true,
30
+ "freq_shift": 0,
31
+ "global_pool_conditions": false,
32
+ "in_channels": 4,
33
+ "layers_per_block": 2,
34
+ "mid_block_scale_factor": 1,
35
+ "norm_eps": 1e-05,
36
+ "norm_num_groups": 32,
37
+ "num_attention_heads": null,
38
+ "num_class_embeds": null,
39
+ "only_cross_attention": false,
40
+ "projection_class_embeddings_input_dim": null,
41
+ "resnet_time_scale_shift": "default",
42
+ "upcast_attention": false,
43
+ "use_linear_projection": false
44
+ }
checkpoint-1500/controlnet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c97c9230a0f69e2b0c2eaf3afc509f45339deef8618c87daf02fd3c5f14dfed
3
+ size 1445259705
checkpoint-1500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f6cf262c0f0f50e6ba3e33e9819a8ef43f543dd1a9eeb3d273c01e4bce7fcd5
3
+ size 725253381
checkpoint-1500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd871114bfbce727438ce23f89347956f1739ce564958f50bdbfe1a07abc7687
3
+ size 14663
checkpoint-1500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:203a72d6c29f42a0e2964fdddc8d7a98df1eccee78fea9de0fa416613390f5c6
3
+ size 557
checkpoint-1500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bab8983b1117c68e4e9a600eacf54bed4c3ac40f58cbf8864e53243107b904fb
3
+ size 563
checkpoint-2000/controlnet/config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "ControlNetModel",
3
+ "_diffusers_version": "0.18.2",
4
+ "act_fn": "silu",
5
+ "attention_head_dim": 8,
6
+ "block_out_channels": [
7
+ 320,
8
+ 640,
9
+ 1280,
10
+ 1280
11
+ ],
12
+ "class_embed_type": null,
13
+ "conditioning_channels": 3,
14
+ "conditioning_embedding_out_channels": [
15
+ 16,
16
+ 32,
17
+ 96,
18
+ 256
19
+ ],
20
+ "controlnet_conditioning_channel_order": "rgb",
21
+ "cross_attention_dim": 768,
22
+ "down_block_types": [
23
+ "CrossAttnDownBlock2D",
24
+ "CrossAttnDownBlock2D",
25
+ "CrossAttnDownBlock2D",
26
+ "DownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "flip_sin_to_cos": true,
30
+ "freq_shift": 0,
31
+ "global_pool_conditions": false,
32
+ "in_channels": 4,
33
+ "layers_per_block": 2,
34
+ "mid_block_scale_factor": 1,
35
+ "norm_eps": 1e-05,
36
+ "norm_num_groups": 32,
37
+ "num_attention_heads": null,
38
+ "num_class_embeds": null,
39
+ "only_cross_attention": false,
40
+ "projection_class_embeddings_input_dim": null,
41
+ "resnet_time_scale_shift": "default",
42
+ "upcast_attention": false,
43
+ "use_linear_projection": false
44
+ }
checkpoint-2000/controlnet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b99398ffce8767ec68007b8d7b502e7d44c7c7333688512401a589db40a294f0
3
+ size 1445259705
checkpoint-2000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a10f34213ce39d3474de8b1ac9856312c24d96d05b8397b2aa4d47ff507fe5be
3
+ size 725253381
checkpoint-2000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7480fbad6b0bfe91c63ebba909fc2844a13eb393b3243d4de47d807338b3503d
3
+ size 14727
checkpoint-2000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd2de9749828adacdf103bf6e9592702bb7067a2c1df27dd62ab38c1eb8c070f
3
+ size 557
checkpoint-2000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dfe94afd21b04fe331bf93a36e6de4869425c4da4a32303493f7f8b96dd6ca61
3
+ size 563
checkpoint-500/controlnet/config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "ControlNetModel",
3
+ "_diffusers_version": "0.18.2",
4
+ "act_fn": "silu",
5
+ "attention_head_dim": 8,
6
+ "block_out_channels": [
7
+ 320,
8
+ 640,
9
+ 1280,
10
+ 1280
11
+ ],
12
+ "class_embed_type": null,
13
+ "conditioning_channels": 3,
14
+ "conditioning_embedding_out_channels": [
15
+ 16,
16
+ 32,
17
+ 96,
18
+ 256
19
+ ],
20
+ "controlnet_conditioning_channel_order": "rgb",
21
+ "cross_attention_dim": 768,
22
+ "down_block_types": [
23
+ "CrossAttnDownBlock2D",
24
+ "CrossAttnDownBlock2D",
25
+ "CrossAttnDownBlock2D",
26
+ "DownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "flip_sin_to_cos": true,
30
+ "freq_shift": 0,
31
+ "global_pool_conditions": false,
32
+ "in_channels": 4,
33
+ "layers_per_block": 2,
34
+ "mid_block_scale_factor": 1,
35
+ "norm_eps": 1e-05,
36
+ "norm_num_groups": 32,
37
+ "num_attention_heads": null,
38
+ "num_class_embeds": null,
39
+ "only_cross_attention": false,
40
+ "projection_class_embeddings_input_dim": null,
41
+ "resnet_time_scale_shift": "default",
42
+ "upcast_attention": false,
43
+ "use_linear_projection": false
44
+ }
checkpoint-500/controlnet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b83a762190307123236cad1204776b9de06816fe7b49dbbd3ddae448367f10d4
3
+ size 1445259705
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4e284caadf8357cee99dac3d226e324a13a10b03decc3312a7f494a9f9e1fe1
3
+ size 725253381
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6fcc47c22ee06812febdd0b0047c232482d4a69648fd25db61061d633d33403f
3
+ size 14663
checkpoint-500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3f196a54202bb4ba1220e8c59f42f9cda0702d68ea83147d814c2fb2f36b8f2
3
+ size 557
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab20ead000e6ae60e9a32d69fb1fb976d10dba2a065da680293a8202053a1975
3
+ size 563
config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "ControlNetModel",
3
+ "_diffusers_version": "0.18.2",
4
+ "act_fn": "silu",
5
+ "attention_head_dim": 8,
6
+ "block_out_channels": [
7
+ 320,
8
+ 640,
9
+ 1280,
10
+ 1280
11
+ ],
12
+ "class_embed_type": null,
13
+ "conditioning_channels": 3,
14
+ "conditioning_embedding_out_channels": [
15
+ 16,
16
+ 32,
17
+ 96,
18
+ 256
19
+ ],
20
+ "controlnet_conditioning_channel_order": "rgb",
21
+ "cross_attention_dim": 768,
22
+ "down_block_types": [
23
+ "CrossAttnDownBlock2D",
24
+ "CrossAttnDownBlock2D",
25
+ "CrossAttnDownBlock2D",
26
+ "DownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "flip_sin_to_cos": true,
30
+ "freq_shift": 0,
31
+ "global_pool_conditions": false,
32
+ "in_channels": 4,
33
+ "layers_per_block": 2,
34
+ "mid_block_scale_factor": 1,
35
+ "norm_eps": 1e-05,
36
+ "norm_num_groups": 32,
37
+ "num_attention_heads": null,
38
+ "num_class_embeds": null,
39
+ "only_cross_attention": false,
40
+ "projection_class_embeddings_input_dim": null,
41
+ "resnet_time_scale_shift": "default",
42
+ "upcast_attention": false,
43
+ "use_linear_projection": false
44
+ }
diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b99398ffce8767ec68007b8d7b502e7d44c7c7333688512401a589db40a294f0
3
+ size 1445259705
image_control.png ADDED
images_0.png ADDED
logs/train_controlnet/1698136955.89282/events.out.tfevents.1698136955.285d4da4d3dd.1278.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3670cab5fe30524bef773a7e8b4ffb31bf6d725f0001d47a11242056689260e0
3
+ size 2284
logs/train_controlnet/1698136955.8950262/hparams.yml ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ caption_column: text
8
+ checkpointing_steps: 500
9
+ checkpoints_total_limit: null
10
+ conditioning_image_column: conditioning_image
11
+ controlnet_model_name_or_path: null
12
+ dataloader_num_workers: 0
13
+ dataset_config_name: null
14
+ dataset_name: Bellaaazzzzz/X_ray
15
+ enable_xformers_memory_efficient_attention: false
16
+ gradient_accumulation_steps: 1
17
+ gradient_checkpointing: false
18
+ hub_model_id: null
19
+ hub_token: null
20
+ image_column: image
21
+ learning_rate: 1.0e-05
22
+ logging_dir: logs
23
+ lr_num_cycles: 1
24
+ lr_power: 1.0
25
+ lr_scheduler: constant
26
+ lr_warmup_steps: 500
27
+ max_grad_norm: 1.0
28
+ max_train_samples: null
29
+ max_train_steps: 2000
30
+ mixed_precision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 1
33
+ output_dir: controlnet-X-ray
34
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
35
+ proportion_empty_prompts: 0
36
+ push_to_hub: true
37
+ report_to: tensorboard
38
+ resolution: 512
39
+ resume_from_checkpoint: null
40
+ revision: null
41
+ scale_lr: false
42
+ seed: null
43
+ set_grads_to_none: false
44
+ tokenizer_name: null
45
+ tracker_project_name: train_controlnet
46
+ train_batch_size: 1
47
+ train_data_dir: null
48
+ use_8bit_adam: false
49
+ validation_steps: 100
logs/train_controlnet/1698137667.7316034/events.out.tfevents.1698137667.285d4da4d3dd.5335.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fbe34d53444a16128f955c2270ed80e6a176d7d50d8f1c738547968f4722868c
3
+ size 2284
logs/train_controlnet/1698137667.7340715/hparams.yml ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ caption_column: text
8
+ checkpointing_steps: 500
9
+ checkpoints_total_limit: null
10
+ conditioning_image_column: conditioning_image
11
+ controlnet_model_name_or_path: null
12
+ dataloader_num_workers: 0
13
+ dataset_config_name: null
14
+ dataset_name: Bellaaazzzzz/X_ray
15
+ enable_xformers_memory_efficient_attention: false
16
+ gradient_accumulation_steps: 1
17
+ gradient_checkpointing: false
18
+ hub_model_id: null
19
+ hub_token: null
20
+ image_column: image
21
+ learning_rate: 1.0e-05
22
+ logging_dir: logs
23
+ lr_num_cycles: 1
24
+ lr_power: 1.0
25
+ lr_scheduler: constant
26
+ lr_warmup_steps: 500
27
+ max_grad_norm: 1.0
28
+ max_train_samples: null
29
+ max_train_steps: 2000
30
+ mixed_precision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 1
33
+ output_dir: controlnet-X-ray
34
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
35
+ proportion_empty_prompts: 0
36
+ push_to_hub: true
37
+ report_to: tensorboard
38
+ resolution: 512
39
+ resume_from_checkpoint: null
40
+ revision: null
41
+ scale_lr: false
42
+ seed: null
43
+ set_grads_to_none: false
44
+ tokenizer_name: null
45
+ tracker_project_name: train_controlnet
46
+ train_batch_size: 1
47
+ train_data_dir: null
48
+ use_8bit_adam: false
49
+ validation_steps: 100
logs/train_controlnet/1698138332.2972698/events.out.tfevents.1698138332.285d4da4d3dd.8317.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23374a0185ded7ef3c4982c4a67b46a1be531e2fd05255a05f4a44e646b12321
3
+ size 2284
logs/train_controlnet/1698138332.2989447/hparams.yml ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ caption_column: text
8
+ checkpointing_steps: 500
9
+ checkpoints_total_limit: null
10
+ conditioning_image_column: conditioning_image
11
+ controlnet_model_name_or_path: null
12
+ dataloader_num_workers: 0
13
+ dataset_config_name: null
14
+ dataset_name: Bellaaazzzzz/X_ray
15
+ enable_xformers_memory_efficient_attention: false
16
+ gradient_accumulation_steps: 1
17
+ gradient_checkpointing: false
18
+ hub_model_id: null
19
+ hub_token: null
20
+ image_column: image
21
+ learning_rate: 1.0e-05
22
+ logging_dir: logs
23
+ lr_num_cycles: 1
24
+ lr_power: 1.0
25
+ lr_scheduler: constant
26
+ lr_warmup_steps: 500
27
+ max_grad_norm: 1.0
28
+ max_train_samples: null
29
+ max_train_steps: 2000
30
+ mixed_precision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 1
33
+ output_dir: controlnet-X-ray
34
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
35
+ proportion_empty_prompts: 0
36
+ push_to_hub: true
37
+ report_to: tensorboard
38
+ resolution: 512
39
+ resume_from_checkpoint: null
40
+ revision: null
41
+ scale_lr: false
42
+ seed: null
43
+ set_grads_to_none: false
44
+ tokenizer_name: null
45
+ tracker_project_name: train_controlnet
46
+ train_batch_size: 1
47
+ train_data_dir: null
48
+ use_8bit_adam: false
49
+ validation_steps: 100
logs/train_controlnet/1698139447.0913413/events.out.tfevents.1698139447.285d4da4d3dd.13142.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1492285252834be05f29224e6b999a698487966b807062e68af5841426261d72
3
+ size 2284
logs/train_controlnet/1698139447.094729/hparams.yml ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ caption_column: text
8
+ checkpointing_steps: 500
9
+ checkpoints_total_limit: null
10
+ conditioning_image_column: conditioning_image
11
+ controlnet_model_name_or_path: null
12
+ dataloader_num_workers: 0
13
+ dataset_config_name: null
14
+ dataset_name: Bellaaazzzzz/X_ray
15
+ enable_xformers_memory_efficient_attention: true
16
+ gradient_accumulation_steps: 4
17
+ gradient_checkpointing: true
18
+ hub_model_id: null
19
+ hub_token: null
20
+ image_column: image
21
+ learning_rate: 1.0e-05
22
+ logging_dir: logs
23
+ lr_num_cycles: 1
24
+ lr_power: 1.0
25
+ lr_scheduler: constant
26
+ lr_warmup_steps: 500
27
+ max_grad_norm: 1.0
28
+ max_train_samples: null
29
+ max_train_steps: 2000
30
+ mixed_precision: null
31
+ num_train_epochs: 2
32
+ num_validation_images: 1
33
+ output_dir: controlnet-X-ray
34
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
35
+ proportion_empty_prompts: 0
36
+ push_to_hub: true
37
+ report_to: tensorboard
38
+ resolution: 512
39
+ resume_from_checkpoint: null
40
+ revision: null
41
+ scale_lr: false
42
+ seed: null
43
+ set_grads_to_none: true
44
+ tokenizer_name: null
45
+ tracker_project_name: train_controlnet
46
+ train_batch_size: 1
47
+ train_data_dir: null
48
+ use_8bit_adam: true
49
+ validation_steps: 100
logs/train_controlnet/events.out.tfevents.1698136955.285d4da4d3dd.1278.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:01ab0b7ccdb54313b4d3bf334ab9c135e4307f8d19cc5d4a92ea414a879a9125
3
+ size 88
logs/train_controlnet/events.out.tfevents.1698137667.285d4da4d3dd.5335.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c51c8b780f3a578a8c2569e1d245a021bf7a926d668ce429fb4e67b26368303
3
+ size 88
logs/train_controlnet/events.out.tfevents.1698138332.285d4da4d3dd.8317.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d3a6aed414f04045f93c2c274d41e5f1ff28d12354d6f091ce1d244ae3eb71ad
3
+ size 88
logs/train_controlnet/events.out.tfevents.1698139447.285d4da4d3dd.13142.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:682e8412d102b8403721572600c5ca0036cf0ef01697eec89922d92b67650f5d
3
+ size 9843962