Hugging Face's logo

ACROSS-Lab
/
PromptTo3D_sd_finetuned

Text-to-Image
Diffusers
TensorBoard
Safetensors
stable-diffusion
stable-diffusion-diffusers
diffusers-training
lora
Model card Files Metrics Community
remi349 commited on
Commit
7dc84d4
•
1 Parent(s): 9d03173

End of training

Browse files
Files changed (32) hide show
  1. README.md +39 -0
  2. checkpoint-51490/config.json +73 -0
  3. checkpoint-51490/model.safetensors +3 -0
  4. checkpoint-51490/optimizer.bin +3 -0
  5. checkpoint-51490/pytorch_lora_weights.safetensors +3 -0
  6. checkpoint-51490/random_states_0.pkl +3 -0
  7. checkpoint-51490/scheduler.bin +3 -0
  8. logs/text2image-fine-tune/1720494219.65187/events.out.tfevents.1720494219.godzilla.50373.1 +3 -0
  9. logs/text2image-fine-tune/1720494219.652725/hparams.yml +48 -0
  10. logs/text2image-fine-tune/1720494470.3701246/events.out.tfevents.1720494470.godzilla.51226.1 +3 -0
  11. logs/text2image-fine-tune/1720494470.3710759/hparams.yml +48 -0
  12. logs/text2image-fine-tune/1720496231.8115919/events.out.tfevents.1720496231.godzilla.57729.1 +3 -0
  13. logs/text2image-fine-tune/1720496231.81269/hparams.yml +48 -0
  14. logs/text2image-fine-tune/1720516712.2810907/events.out.tfevents.1720516712.godzilla.177716.1 +3 -0
  15. logs/text2image-fine-tune/1720516712.2820623/hparams.yml +48 -0
  16. logs/text2image-fine-tune/1720672534.1155264/events.out.tfevents.1720672534.godzilla.372722.1 +3 -0
  17. logs/text2image-fine-tune/1720672534.1165068/hparams.yml +48 -0
  18. logs/text2image-fine-tune/1720684533.528313/events.out.tfevents.1720684533.godzilla.472177.1 +3 -0
  19. logs/text2image-fine-tune/1720684533.5295682/hparams.yml +48 -0
  20. logs/text2image-fine-tune/1720684770.8480604/events.out.tfevents.1720684770.godzilla.473711.1 +3 -0
  21. logs/text2image-fine-tune/1720684770.8491058/hparams.yml +48 -0
  22. logs/text2image-fine-tune/1720684829.9974127/events.out.tfevents.1720684829.godzilla.474174.1 +3 -0
  23. logs/text2image-fine-tune/1720684829.9985082/hparams.yml +48 -0
  24. logs/text2image-fine-tune/events.out.tfevents.1720494219.godzilla.50373.0 +3 -0
  25. logs/text2image-fine-tune/events.out.tfevents.1720494470.godzilla.51226.0 +3 -0
  26. logs/text2image-fine-tune/events.out.tfevents.1720496231.godzilla.57729.0 +3 -0
  27. logs/text2image-fine-tune/events.out.tfevents.1720516712.godzilla.177716.0 +3 -0
  28. logs/text2image-fine-tune/events.out.tfevents.1720672534.godzilla.372722.0 +3 -0
  29. logs/text2image-fine-tune/events.out.tfevents.1720684533.godzilla.472177.0 +3 -0
  30. logs/text2image-fine-tune/events.out.tfevents.1720684770.godzilla.473711.0 +3 -0
  31. logs/text2image-fine-tune/events.out.tfevents.1720684829.godzilla.474174.0 +3 -0
  32. pytorch_lora_weights.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: creativeml-openrail-m
3
+ library_name: diffusers
4
+ tags:
5
+ - stable-diffusion
6
+ - stable-diffusion-diffusers
7
+ - text-to-image
8
+ - diffusers
9
+ - diffusers-training
10
+ - lora
11
+ base_model: stabilityai/stable-diffusion-2-1
12
+ inference: true
13
+ ---
14
+
15
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
16
+ should probably proofread and complete it, then remove this comment. -->
17
+
18
+
19
+ # LoRA text2image fine-tuning - remi349/sd_trained_3D_lora
20
+ These are LoRA adaption weights for stabilityai/stable-diffusion-2-1. The weights were fine-tuned on the remi349/finetuning_dataset_for_3D_training dataset. You can find some example images in the following.
21
+
22
+
23
+
24
+
25
+ ## Intended uses & limitations
26
+
27
+ #### How to use
28
+
29
+ ```python
30
+ # TODO: add an example code snippet for running this diffusion pipeline
31
+ ```
32
+
33
+ #### Limitations and bias
34
+
35
+ [TODO: provide examples of latent issues and potential remediations]
36
+
37
+ ## Training details
38
+
39
+ [TODO: describe the data used to train the model]
checkpoint-51490/config.json ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.30.0.dev0",
4
+ "_name_or_path": "stabilityai/stable-diffusion-2-1",
5
+ "act_fn": "silu",
6
+ "addition_embed_type": null,
7
+ "addition_embed_type_num_heads": 64,
8
+ "addition_time_embed_dim": null,
9
+ "attention_head_dim": [
10
+ 5,
11
+ 10,
12
+ 20,
13
+ 20
14
+ ],
15
+ "attention_type": "default",
16
+ "block_out_channels": [
17
+ 320,
18
+ 640,
19
+ 1280,
20
+ 1280
21
+ ],
22
+ "center_input_sample": false,
23
+ "class_embed_type": null,
24
+ "class_embeddings_concat": false,
25
+ "conv_in_kernel": 3,
26
+ "conv_out_kernel": 3,
27
+ "cross_attention_dim": 1024,
28
+ "cross_attention_norm": null,
29
+ "down_block_types": [
30
+ "CrossAttnDownBlock2D",
31
+ "CrossAttnDownBlock2D",
32
+ "CrossAttnDownBlock2D",
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,
40
+ "flip_sin_to_cos": true,
41
+ "freq_shift": 0,
42
+ "in_channels": 4,
43
+ "layers_per_block": 2,
44
+ "mid_block_only_cross_attention": null,
45
+ "mid_block_scale_factor": 1,
46
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
47
+ "norm_eps": 1e-05,
48
+ "norm_num_groups": 32,
49
+ "num_attention_heads": null,
50
+ "num_class_embeds": null,
51
+ "only_cross_attention": false,
52
+ "out_channels": 4,
53
+ "projection_class_embeddings_input_dim": 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,
61
+ "time_embedding_dim": null,
62
+ "time_embedding_type": "positional",
63
+ "timestep_post_act": null,
64
+ "transformer_layers_per_block": 1,
65
+ "up_block_types": [
66
+ "UpBlock2D",
67
+ "CrossAttnUpBlock2D",
68
+ "CrossAttnUpBlock2D",
69
+ "CrossAttnUpBlock2D"
70
+ ],
71
+ "upcast_attention": true,
72
+ "use_linear_projection": true
73
+ }
checkpoint-51490/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5bed0e69a95183adbbe03dbb1f724208c16e7f984891dbd5e84fbc2f695d5bf6
3
+ size 3467087168
checkpoint-51490/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4eb3dcdf6a3df1b90ac8eb3c3158db87cecdb7b50d6cc7fdca6de0a83a36ee41
3
+ size 6857349
checkpoint-51490/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe5cfc5433c5620d2c272b317d2d8c336b1c125c5944c9e9824c4a99304a9487
3
+ size 3357296
checkpoint-51490/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a6a683f35756f4d0883db43fda473e13b39e236b74245cef3aed23ceef36934a
3
+ size 15691
checkpoint-51490/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ab54fc1b09d963c095b20c1796e31d79aab1a428d6fd18f2e8fdc8a77ed8fca
3
+ size 563
logs/text2image-fine-tune/1720494219.65187/events.out.tfevents.1720494219.godzilla.50373.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9fa640663a5c2a36c97aba5483d76d77b59a835be1c94028336f28c41327b993
3
+ size 2184
logs/text2image-fine-tune/1720494219.652725/hparams.yml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ center_crop: false
9
+ checkpointing_steps: 5000
10
+ checkpoints_total_limit: null
11
+ data_path: ../../../SD/train
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
+ image_column: image
19
+ learning_rate: 0.0001
20
+ local_rank: 0
21
+ logging_dir: logs
22
+ lr_scheduler: constant
23
+ lr_warmup_steps: 0
24
+ max_grad_norm: 1.0
25
+ max_train_samples: null
26
+ max_train_steps: 51500
27
+ mixed_precision: null
28
+ noise_offset: 0
29
+ num_train_epochs: 100
30
+ num_validation_images: 4
31
+ output_dir: sd_trained_3D_lora
32
+ prediction_type: null
33
+ pretrained_model_name: stabilityai/stable-diffusion-2-1
34
+ push_to_hub: true
35
+ random_flip: true
36
+ rank: 4
37
+ report_to: tensorboard
38
+ resolution: 768
39
+ resume_from_checkpoint: ./sd_trained_3D_lora/checkpoint-51490
40
+ revision: null
41
+ scale_lr: false
42
+ seed: 42
43
+ snr_gamma: null
44
+ train_batch_size: 1
45
+ use_8bit_adam: false
46
+ validation_epochs: 1
47
+ validation_prompt: null
48
+ variant: null
logs/text2image-fine-tune/1720494470.3701246/events.out.tfevents.1720494470.godzilla.51226.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba50c5ca05575d22f9cc2f3ff12f11069c41166f81638865889149d268153d6f
3
+ size 2184
logs/text2image-fine-tune/1720494470.3710759/hparams.yml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ center_crop: false
9
+ checkpointing_steps: 5000
10
+ checkpoints_total_limit: null
11
+ data_path: ../../../SD/train
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
+ image_column: image
19
+ learning_rate: 0.0001
20
+ local_rank: 0
21
+ logging_dir: logs
22
+ lr_scheduler: constant
23
+ lr_warmup_steps: 0
24
+ max_grad_norm: 1.0
25
+ max_train_samples: null
26
+ max_train_steps: 51500
27
+ mixed_precision: null
28
+ noise_offset: 0
29
+ num_train_epochs: 100
30
+ num_validation_images: 4
31
+ output_dir: sd_trained_3D_lora
32
+ prediction_type: null
33
+ pretrained_model_name: stabilityai/stable-diffusion-2-1
34
+ push_to_hub: true
35
+ random_flip: true
36
+ rank: 4
37
+ report_to: tensorboard
38
+ resolution: 512
39
+ resume_from_checkpoint: ./sd_trained_3D_lora/checkpoint-51490
40
+ revision: null
41
+ scale_lr: false
42
+ seed: 42
43
+ snr_gamma: null
44
+ train_batch_size: 1
45
+ use_8bit_adam: false
46
+ validation_epochs: 1
47
+ validation_prompt: null
48
+ variant: null
logs/text2image-fine-tune/1720496231.8115919/events.out.tfevents.1720496231.godzilla.57729.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3212941281d301fde9783afb1f8d22365055dd8a68156f2c509156df7a278013
3
+ size 2184
logs/text2image-fine-tune/1720496231.81269/hparams.yml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ center_crop: false
9
+ checkpointing_steps: 5000
10
+ checkpoints_total_limit: null
11
+ data_path: ../../../SD/train
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
+ image_column: image
19
+ learning_rate: 0.0001
20
+ local_rank: 0
21
+ logging_dir: logs
22
+ lr_scheduler: constant
23
+ lr_warmup_steps: 0
24
+ max_grad_norm: 1.0
25
+ max_train_samples: null
26
+ max_train_steps: 51500
27
+ mixed_precision: null
28
+ noise_offset: 0
29
+ num_train_epochs: 100
30
+ num_validation_images: 4
31
+ output_dir: sd_trained_3D_lora
32
+ prediction_type: null
33
+ pretrained_model_name: stabilityai/stable-diffusion-2-1
34
+ push_to_hub: true
35
+ random_flip: true
36
+ rank: 4
37
+ report_to: tensorboard
38
+ resolution: 512
39
+ resume_from_checkpoint: ./sd_trained_3D_lora/checkpoint-51490
40
+ revision: null
41
+ scale_lr: false
42
+ seed: 42
43
+ snr_gamma: null
44
+ train_batch_size: 1
45
+ use_8bit_adam: false
46
+ validation_epochs: 1
47
+ validation_prompt: null
48
+ variant: null
logs/text2image-fine-tune/1720516712.2810907/events.out.tfevents.1720516712.godzilla.177716.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b58c71b29effa5a3619ff3455e1739ea8ad654085a4c6a027966c4c0f866e22
3
+ size 2184
logs/text2image-fine-tune/1720516712.2820623/hparams.yml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ center_crop: false
9
+ checkpointing_steps: 5000
10
+ checkpoints_total_limit: null
11
+ data_path: ../../../SD/train
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
+ image_column: image
19
+ learning_rate: 0.0001
20
+ local_rank: 0
21
+ logging_dir: logs
22
+ lr_scheduler: constant
23
+ lr_warmup_steps: 0
24
+ max_grad_norm: 1.0
25
+ max_train_samples: null
26
+ max_train_steps: 51500
27
+ mixed_precision: null
28
+ noise_offset: 0
29
+ num_train_epochs: 100
30
+ num_validation_images: 4
31
+ output_dir: sd_trained_3D_lora
32
+ prediction_type: null
33
+ pretrained_model_name: stabilityai/stable-diffusion-2-1
34
+ push_to_hub: true
35
+ random_flip: true
36
+ rank: 4
37
+ report_to: tensorboard
38
+ resolution: 512
39
+ resume_from_checkpoint: ./sd_trained_3D_lora/checkpoint-51490
40
+ revision: null
41
+ scale_lr: false
42
+ seed: 42
43
+ snr_gamma: null
44
+ train_batch_size: 1
45
+ use_8bit_adam: false
46
+ validation_epochs: 1
47
+ validation_prompt: null
48
+ variant: null
logs/text2image-fine-tune/1720672534.1155264/events.out.tfevents.1720672534.godzilla.372722.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:463f82f4c8bb875dcb4fa6f8fd2c5902ce87c8376424dcdd6351382baeaad9fd
3
+ size 2184
logs/text2image-fine-tune/1720672534.1165068/hparams.yml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ center_crop: false
9
+ checkpointing_steps: 5000
10
+ checkpoints_total_limit: null
11
+ data_path: ../../../SD/train
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
+ image_column: image
19
+ learning_rate: 0.0001
20
+ local_rank: 0
21
+ logging_dir: logs
22
+ lr_scheduler: constant
23
+ lr_warmup_steps: 0
24
+ max_grad_norm: 1.0
25
+ max_train_samples: null
26
+ max_train_steps: 51500
27
+ mixed_precision: null
28
+ noise_offset: 0
29
+ num_train_epochs: 100
30
+ num_validation_images: 4
31
+ output_dir: sd_trained_3D_lora
32
+ prediction_type: null
33
+ pretrained_model_name: stabilityai/stable-diffusion-2-1
34
+ push_to_hub: true
35
+ random_flip: true
36
+ rank: 4
37
+ report_to: tensorboard
38
+ resolution: 512
39
+ resume_from_checkpoint: ./sd_trained_3D_lora/checkpoint-51490
40
+ revision: null
41
+ scale_lr: false
42
+ seed: 42
43
+ snr_gamma: null
44
+ train_batch_size: 1
45
+ use_8bit_adam: false
46
+ validation_epochs: 1
47
+ validation_prompt: null
48
+ variant: null
logs/text2image-fine-tune/1720684533.528313/events.out.tfevents.1720684533.godzilla.472177.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e64778b7f0cc1a30fbf64370c4ec496e3b20d827b7010bff8149ade4f8fe750
3
+ size 2184
logs/text2image-fine-tune/1720684533.5295682/hparams.yml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ center_crop: false
9
+ checkpointing_steps: 5000
10
+ checkpoints_total_limit: null
11
+ data_path: ../../../SD/train
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
+ image_column: image
19
+ learning_rate: 0.0001
20
+ local_rank: 0
21
+ logging_dir: logs
22
+ lr_scheduler: constant
23
+ lr_warmup_steps: 0
24
+ max_grad_norm: 1.0
25
+ max_train_samples: null
26
+ max_train_steps: 51500
27
+ mixed_precision: null
28
+ noise_offset: 0
29
+ num_train_epochs: 100
30
+ num_validation_images: 4
31
+ output_dir: sd_trained_3D_lora
32
+ prediction_type: null
33
+ pretrained_model_name: stabilityai/stable-diffusion-2-1
34
+ push_to_hub: true
35
+ random_flip: true
36
+ rank: 4
37
+ report_to: tensorboard
38
+ resolution: 512
39
+ resume_from_checkpoint: ./sd_trained_3D_lora/checkpoint-51490
40
+ revision: null
41
+ scale_lr: false
42
+ seed: 42
43
+ snr_gamma: null
44
+ train_batch_size: 1
45
+ use_8bit_adam: false
46
+ validation_epochs: 1
47
+ validation_prompt: null
48
+ variant: null
logs/text2image-fine-tune/1720684770.8480604/events.out.tfevents.1720684770.godzilla.473711.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aafc257e914989b6e3cd5b8f3593e7710706e7c8c7a68b91f5bd6bb52469dd0a
3
+ size 2177
logs/text2image-fine-tune/1720684770.8491058/hparams.yml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ center_crop: false
9
+ checkpointing_steps: 5000
10
+ checkpoints_total_limit: null
11
+ data_path: ../../../SD/train
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
+ image_column: image
19
+ learning_rate: 0.0001
20
+ local_rank: 0
21
+ logging_dir: logs
22
+ lr_scheduler: constant
23
+ lr_warmup_steps: 0
24
+ max_grad_norm: 1.0
25
+ max_train_samples: null
26
+ max_train_steps: 51500
27
+ mixed_precision: null
28
+ noise_offset: 0
29
+ num_train_epochs: 100
30
+ num_validation_images: 4
31
+ output_dir: sd_trained_3D_lora
32
+ prediction_type: null
33
+ pretrained_model_name: stabilityai/stable-diffusion-2-1
34
+ push_to_hub: true
35
+ random_flip: true
36
+ rank: 4
37
+ report_to: tensorboard
38
+ resolution: 512
39
+ resume_from_checkpoint: ./sd_trained_3D_lora/SD_for_3D
40
+ revision: null
41
+ scale_lr: false
42
+ seed: 42
43
+ snr_gamma: null
44
+ train_batch_size: 1
45
+ use_8bit_adam: false
46
+ validation_epochs: 1
47
+ validation_prompt: null
48
+ variant: null
logs/text2image-fine-tune/1720684829.9974127/events.out.tfevents.1720684829.godzilla.474174.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:48d75bbef43e956def2865396cd8e7a58c1bf4a5a1fb9c9843ec670022d0a556
3
+ size 2184
logs/text2image-fine-tune/1720684829.9985082/hparams.yml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ center_crop: false
9
+ checkpointing_steps: 5000
10
+ checkpoints_total_limit: null
11
+ data_path: ../../../SD/train
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
+ image_column: image
19
+ learning_rate: 0.0001
20
+ local_rank: 0
21
+ logging_dir: logs
22
+ lr_scheduler: constant
23
+ lr_warmup_steps: 0
24
+ max_grad_norm: 1.0
25
+ max_train_samples: null
26
+ max_train_steps: 51500
27
+ mixed_precision: null
28
+ noise_offset: 0
29
+ num_train_epochs: 100
30
+ num_validation_images: 4
31
+ output_dir: sd_trained_3D_lora
32
+ prediction_type: null
33
+ pretrained_model_name: stabilityai/stable-diffusion-2-1
34
+ push_to_hub: true
35
+ random_flip: true
36
+ rank: 4
37
+ report_to: tensorboard
38
+ resolution: 512
39
+ resume_from_checkpoint: ./sd_trained_3D_lora/checkpoint-51490
40
+ revision: null
41
+ scale_lr: false
42
+ seed: 42
43
+ snr_gamma: null
44
+ train_batch_size: 1
45
+ use_8bit_adam: false
46
+ validation_epochs: 1
47
+ validation_prompt: null
48
+ variant: null
logs/text2image-fine-tune/events.out.tfevents.1720494219.godzilla.50373.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:266abf0b427634d3ed2d7ed44924aa088d717b78ab80e3c0b3a7dadcc570845e
3
+ size 40
logs/text2image-fine-tune/events.out.tfevents.1720494470.godzilla.51226.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2bf98a38fac7123e1cdf263253b813dc7925db45f7d330b2f167f30d099a8c38
3
+ size 540
logs/text2image-fine-tune/events.out.tfevents.1720496231.godzilla.57729.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e15adc97d71ebc391ce030a219ce21c045dc69964bbb26a356e1d8a23fa517a3
3
+ size 540
logs/text2image-fine-tune/events.out.tfevents.1720516712.godzilla.177716.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:442658d6a3a9ce9668213eb005e452b1e0c7585c03fe6de99497b2dcfd0322bc
3
+ size 540
logs/text2image-fine-tune/events.out.tfevents.1720672534.godzilla.372722.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eff95198c8c7902d20dbecd6fb7b5bbe99ae6ba79d167b4a27e66183c7e80735
3
+ size 540
logs/text2image-fine-tune/events.out.tfevents.1720684533.godzilla.472177.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:06ed6566debcaed8199b013f0e5b50a97f5038c5b54abd761a2e71b063e22486
3
+ size 540
logs/text2image-fine-tune/events.out.tfevents.1720684770.godzilla.473711.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2dc118d2fd9c099ffab0aeb90771185bf5233d5729226c6b7c0a6562cdfe8183
3
+ size 40
logs/text2image-fine-tune/events.out.tfevents.1720684829.godzilla.474174.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:592689af5a8a2c644eb6045184402242c1adc548100854b3cabbafa1a775990a
3
+ size 540
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2991cbe6ab8d5620fd33567a5f6cadcf517197d10c905391a3306229e59b53b4
3
+ size 3357296