arvnoodle commited on
Commit
9bb80b8
1 Parent(s): b5262cb

End of training

Browse files
.gitattributes CHANGED
@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ image_0.png filter=lfs diff=lfs merge=lfs -text
37
+ image_1.png filter=lfs diff=lfs merge=lfs -text
38
+ image_2.png filter=lfs diff=lfs merge=lfs -text
39
+ image_3.png filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: stabilityai/stable-diffusion-xl-base-1.0
3
+ library_name: diffusers
4
+ license: openrail++
5
+ tags:
6
+ - text-to-image
7
+ - text-to-image
8
+ - diffusers-training
9
+ - diffusers
10
+ - dora
11
+ - template:sd-lora
12
+ - stable-diffusion-xl
13
+ - stable-diffusion-xl-diffusers
14
+ instance_prompt: a photo of sks dog
15
+ widget:
16
+ - text: A photo of sks dog in a bucket
17
+ output:
18
+ url: image_0.png
19
+ - text: A photo of sks dog in a bucket
20
+ output:
21
+ url: image_1.png
22
+ - text: A photo of sks dog in a bucket
23
+ output:
24
+ url: image_2.png
25
+ - text: A photo of sks dog in a bucket
26
+ output:
27
+ url: image_3.png
28
+ ---
29
+
30
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
31
+ should probably proofread and complete it, then remove this comment. -->
32
+
33
+
34
+ # SDXL LoRA DreamBooth - arvnoodle/lora-trained-xl
35
+
36
+ <Gallery />
37
+
38
+ ## Model description
39
+
40
+ These are arvnoodle/lora-trained-xl LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0.
41
+
42
+ The weights were trained using [DreamBooth](https://dreambooth.github.io/).
43
+
44
+ LoRA for the text encoder was enabled: False.
45
+
46
+ Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
47
+
48
+ ## Trigger words
49
+
50
+ You should use a photo of sks dog to trigger the image generation.
51
+
52
+ ## Download model
53
+
54
+ Weights for this model are available in Safetensors format.
55
+
56
+ [Download](arvnoodle/lora-trained-xl/tree/main) them in the Files & versions tab.
57
+
58
+
59
+
60
+ ## Intended uses & limitations
61
+
62
+ #### How to use
63
+
64
+ ```python
65
+ # TODO: add an example code snippet for running this diffusion pipeline
66
+ ```
67
+
68
+ #### Limitations and bias
69
+
70
+ [TODO: provide examples of latent issues and potential remediations]
71
+
72
+ ## Training details
73
+
74
+ [TODO: describe the data used to train the model]
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e441e1d3d7d90a2d92dce0d4b6595904f9445d01866ffd579f64a3b40d0816c4
3
+ size 53185858
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7961320722839f21b4b0d4b5c1868839e314e787fbb0a36c508675d98281afc1
3
+ size 26138248
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6acba21ec6bbe568a963b7a35615010ed03fcce05bf37c195d12ccba089130c2
3
+ size 14344
checkpoint-500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18b984273ea2d45b7ffb1d047bb359d93111e41fcad70d16a1b453fd38f72636
3
+ size 988
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6cf34ac8dd2addc454aeb1ee4eaba3267c2841857e117be40ac2ed02cf581c9c
3
+ size 1000
image_0.png ADDED

Git LFS Details

  • SHA256: 4ae65de1220104facc707776b2223ff5ced2d65455155d552e678a87f287a87f
  • Pointer size: 132 Bytes
  • Size of remote file: 1.6 MB
image_1.png ADDED

Git LFS Details

  • SHA256: e1a8081b7965be85ac014cfc37c78f0e1bfafbaabb8274697444fdd713e1e756
  • Pointer size: 132 Bytes
  • Size of remote file: 1.51 MB
image_2.png ADDED

Git LFS Details

  • SHA256: a097afde0682fed773b3d23beb2da4af2c04a3f825660567f627ab833826d043
  • Pointer size: 132 Bytes
  • Size of remote file: 1.58 MB
image_3.png ADDED

Git LFS Details

  • SHA256: b94eb29465326ec23e3cc7c54c58b854a3231f25ed9ecd095d21633e659c1163
  • Pointer size: 132 Bytes
  • Size of remote file: 1.49 MB
logs/dreambooth-lora-sd-xl/1723700823.133589/events.out.tfevents.1723700823.c143be952141.3104.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:db3f230bca1ee49ae876e22a79aa3d130ae0e5e015a12cbdcae21285bc5b615f
3
+ size 3297
logs/dreambooth-lora-sd-xl/1723700823.1429842/hparams.yml ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.0001
5
+ adam_weight_decay_text_encoder: 0.001
6
+ allow_tf32: false
7
+ cache_dir: null
8
+ caption_column: null
9
+ center_crop: false
10
+ checkpointing_steps: 500
11
+ checkpoints_total_limit: null
12
+ class_data_dir: null
13
+ class_prompt: null
14
+ dataloader_num_workers: 0
15
+ dataset_config_name: null
16
+ dataset_name: null
17
+ do_edm_style_training: false
18
+ enable_xformers_memory_efficient_attention: false
19
+ gradient_accumulation_steps: 4
20
+ gradient_checkpointing: false
21
+ hub_model_id: null
22
+ hub_token: null
23
+ image_column: image
24
+ instance_data_dir: dog
25
+ instance_prompt: a photo of sks dog
26
+ learning_rate: 0.0001
27
+ local_rank: -1
28
+ logging_dir: logs
29
+ lr_num_cycles: 1
30
+ lr_power: 1.0
31
+ lr_scheduler: constant
32
+ lr_warmup_steps: 0
33
+ max_grad_norm: 1.0
34
+ max_train_steps: 500
35
+ mixed_precision: fp16
36
+ num_class_images: 100
37
+ num_train_epochs: 250
38
+ num_validation_images: 4
39
+ optimizer: AdamW
40
+ output_dir: lora-trained-xl
41
+ output_kohya_format: false
42
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
43
+ pretrained_vae_model_name_or_path: madebyollin/sdxl-vae-fp16-fix
44
+ prior_generation_precision: null
45
+ prior_loss_weight: 1.0
46
+ prodigy_beta3: null
47
+ prodigy_decouple: true
48
+ prodigy_safeguard_warmup: true
49
+ prodigy_use_bias_correction: true
50
+ push_to_hub: true
51
+ random_flip: false
52
+ rank: 4
53
+ repeats: 1
54
+ report_to: tensorboard
55
+ resolution: 1024
56
+ resume_from_checkpoint: null
57
+ revision: null
58
+ sample_batch_size: 4
59
+ scale_lr: false
60
+ seed: 0
61
+ snr_gamma: null
62
+ text_encoder_lr: 5.0e-06
63
+ train_batch_size: 1
64
+ train_text_encoder: false
65
+ use_8bit_adam: false
66
+ use_dora: true
67
+ validation_epochs: 25
68
+ validation_prompt: A photo of sks dog in a bucket
69
+ variant: null
70
+ with_prior_preservation: false
logs/dreambooth-lora-sd-xl/events.out.tfevents.1723700823.c143be952141.3104.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4aa9f24764a3c247346c4657db0a7c5b9c58718620d477f4d7e2b8edd7be729
3
+ size 66747381
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7961320722839f21b4b0d4b5c1868839e314e787fbb0a36c508675d98281afc1
3
+ size 26138248