hunz commited on
Commit
ae4a655
·
1 Parent(s): db5ad94

End of training

Browse files
README.md CHANGED
@@ -1,3 +1,23 @@
 
1
  ---
2
- license: openrail
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
1
+
2
  ---
3
+ license: openrail++
4
+ base_model: stabilityai/stable-diffusion-xl-base-1.0
5
+ instance_prompt: fire
6
+ tags:
7
+ - stable-diffusion-xl
8
+ - stable-diffusion-xl-diffusers
9
+ - text-to-image
10
+ - diffusers
11
+ - lora
12
+ inference: true
13
  ---
14
+
15
+ # LoRA DreamBooth - hunz/stable-diffusion-xl-fire
16
+
17
+ These are LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0. The weights were trained on fire using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
18
+
19
+
20
+
21
+ LoRA for the text encoder was enabled: False.
22
+
23
+ Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
logs/dreambooth-lora-sd-xl/1694525566.2835274/events.out.tfevents.1694525566.brainai.7333.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5baa416b9e8752d5a69cba54e8ef4ae46c63194381afdaad42ae53ff4cc88e0
3
+ size 2655
logs/dreambooth-lora-sd-xl/1694525566.2866879/hparams.yml ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 717
8
+ checkpoints_total_limit: null
9
+ class_data_dir: null
10
+ class_prompt: null
11
+ crops_coords_top_left_h: 0
12
+ crops_coords_top_left_w: 0
13
+ dataloader_num_workers: 0
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 2
16
+ gradient_checkpointing: false
17
+ hub_model_id: null
18
+ hub_token: null
19
+ instance_data_dir: iron
20
+ instance_prompt: fire
21
+ learning_rate: 0.0001
22
+ local_rank: 0
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: 500
30
+ mixed_precision: fp16
31
+ num_class_images: 100
32
+ num_train_epochs: 500
33
+ num_validation_images: 4
34
+ output_dir: stable-diffusion-xl-fire
35
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
36
+ pretrained_vae_model_name_or_path: madebyollin/sdxl-vae-fp16-fix
37
+ prior_generation_precision: null
38
+ prior_loss_weight: 1.0
39
+ push_to_hub: true
40
+ rank: 4
41
+ report_to: tensorboard
42
+ resolution: 1024
43
+ resume_from_checkpoint: null
44
+ revision: null
45
+ sample_batch_size: 4
46
+ scale_lr: false
47
+ seed: 0
48
+ train_batch_size: 2
49
+ train_text_encoder: false
50
+ use_8bit_adam: false
51
+ validation_epochs: 50
52
+ validation_prompt: null
53
+ with_prior_preservation: false
logs/dreambooth-lora-sd-xl/1694571400.486743/events.out.tfevents.1694571400.brainai.13433.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d555864e477318feb2a988fca9290d893d5483af3a582fa1665617edb53be723
3
+ size 2659
logs/dreambooth-lora-sd-xl/1694571400.4894001/hparams.yml ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 717
8
+ checkpoints_total_limit: null
9
+ class_data_dir: null
10
+ class_prompt: null
11
+ crops_coords_top_left_h: 0
12
+ crops_coords_top_left_w: 0
13
+ dataloader_num_workers: 0
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 2
16
+ gradient_checkpointing: false
17
+ hub_model_id: null
18
+ hub_token: null
19
+ instance_data_dir: wildfire
20
+ instance_prompt: fire
21
+ learning_rate: 0.0001
22
+ local_rank: 0
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: 500
30
+ mixed_precision: fp16
31
+ num_class_images: 100
32
+ num_train_epochs: 500
33
+ num_validation_images: 4
34
+ output_dir: stable-diffusion-xl-fire
35
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
36
+ pretrained_vae_model_name_or_path: madebyollin/sdxl-vae-fp16-fix
37
+ prior_generation_precision: null
38
+ prior_loss_weight: 1.0
39
+ push_to_hub: true
40
+ rank: 4
41
+ report_to: tensorboard
42
+ resolution: 1024
43
+ resume_from_checkpoint: null
44
+ revision: null
45
+ sample_batch_size: 4
46
+ scale_lr: false
47
+ seed: 0
48
+ train_batch_size: 2
49
+ train_text_encoder: false
50
+ use_8bit_adam: false
51
+ validation_epochs: 50
52
+ validation_prompt: null
53
+ with_prior_preservation: false
logs/dreambooth-lora-sd-xl/events.out.tfevents.1694525566.brainai.7333.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:784a61d27f557a27fe5a6df849c46405417efc8bcf41b542a4747be6b6bd621b
3
+ size 83526
logs/dreambooth-lora-sd-xl/events.out.tfevents.1694571400.brainai.13433.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ed6af7b79213a4e1a4bbeefcb61b269fc3fcebd46349666ab8c67138a8245c30
3
+ size 83526
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:462799f7398d3ffeabe612bd719ff56687d8f593b1c627b9708e9cc4f5f335a2
3
+ size 23401064