arvnoodle commited on
Commit
4c43764
1 Parent(s): cc77184

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
+ - lora
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-128
35
+
36
+ <Gallery />
37
+
38
+ ## Model description
39
+
40
+ These are arvnoodle/lora-trained-xl-128 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-128/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:6281736f2154cbfa355bec66ce081d1281e425e43c8989892313ca2bcba36d91
3
+ size 1487305202
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:674a5c8bbb7e5b56ffea2d5b9408673e29d591b2c4dbd3abc7b17778fa58479b
3
+ size 743349552
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1591d4af518f754596de5d4650a5ca84562a4cb7576178e3a8680dcaa9fe0bec
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: 2e350613c65ffa8b310f7084a001f6442b5cbfc13167139c693fcd5456122503
  • Pointer size: 132 Bytes
  • Size of remote file: 1.66 MB
image_1.png ADDED

Git LFS Details

  • SHA256: 62784b39060eba00e63338da1dd79ed9da81a0c8d694a4864843956b0e3e0510
  • Pointer size: 132 Bytes
  • Size of remote file: 1.52 MB
image_2.png ADDED

Git LFS Details

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

Git LFS Details

  • SHA256: 3f042247d1b74b846b9b2367608b0aa466a13bf696f26b0b5a21e401d039c42b
  • Pointer size: 132 Bytes
  • Size of remote file: 1.56 MB
logs/dreambooth-lora-sd-xl/1724201861.9796522/events.out.tfevents.1724201861.e518a0cbb89a.2460.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:065d5624172ad1435226efeac651531adc55d360cbf385c5bbe5e220c13ae043
3
+ size 3301
logs/dreambooth-lora-sd-xl/1724201861.9812646/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-128
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: 128
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: false
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.1724201861.e518a0cbb89a.2460.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:920d75d510df149d877864696f6d618a0dfe5cc8d0836516a38c61942f6789e2
3
+ size 68119699
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:674a5c8bbb7e5b56ffea2d5b9408673e29d591b2c4dbd3abc7b17778fa58479b
3
+ size 743349552