bgr2rgb commited on
Commit
6edbafd
1 Parent(s): 2ebb47e

End of training

Browse files
README.md ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: creativeml-openrail-m
3
+ library_name: diffusers
4
+ tags:
5
+ - text-to-image
6
+ - diffusers
7
+ - lora
8
+ - stable-diffusion
9
+ - stable-diffusion-diffusers
10
+ inference: true
11
+ base_model: CompVis/stable-diffusion-v1-4
12
+ instance_prompt: a photo of sks dog
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 DreamBooth - bgr2rgb/output
20
+
21
+ These are LoRA adaption weights for CompVis/stable-diffusion-v1-4. The weights were trained on a photo of sks dog using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
22
+
23
+
24
+
25
+ LoRA for the text encoder was enabled: False.
26
+
27
+
28
+ ## Intended uses & limitations
29
+
30
+ #### How to use
31
+
32
+ ```python
33
+ # TODO: add an example code snippet for running this diffusion pipeline
34
+ ```
35
+
36
+ #### Limitations and bias
37
+
38
+ [TODO: provide examples of latent issues and potential remediations]
39
+
40
+ ## Training details
41
+
42
+ [TODO: describe the data used to train the model]
logs/dreambooth-lora/1708859151.8424776/events.out.tfevents.1708859151.greengpu.greengpu.com.1281380.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:41e4f1436edb18162be539b613787cfc249def74c127ecc3a00afca8e4612db1
3
+ size 2631
logs/dreambooth-lora/1708859151.8454983/hparams.yml ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 500
8
+ checkpoints_total_limit: null
9
+ class_data_dir: class_images
10
+ class_labels_conditioning: null
11
+ class_prompt: a photo of dog
12
+ dataloader_num_workers: 0
13
+ enable_xformers_memory_efficient_attention: false
14
+ gradient_accumulation_steps: 2
15
+ gradient_checkpointing: true
16
+ hub_model_id: null
17
+ hub_token: null
18
+ instance_data_dir: dog
19
+ instance_prompt: a photo of sks dog
20
+ learning_rate: 5.0e-06
21
+ local_rank: -1
22
+ logging_dir: logs
23
+ lr_num_cycles: 1
24
+ lr_power: 1.0
25
+ lr_scheduler: constant
26
+ lr_warmup_steps: 0
27
+ max_grad_norm: 1.0
28
+ max_train_steps: 400
29
+ mixed_precision: null
30
+ num_class_images: 100
31
+ num_train_epochs: 8
32
+ num_validation_images: 4
33
+ output_dir: output
34
+ pre_compute_text_embeddings: false
35
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
36
+ prior_generation_precision: null
37
+ prior_loss_weight: 1.0
38
+ push_to_hub: true
39
+ rank: 4
40
+ report_to: tensorboard
41
+ resolution: 512
42
+ resume_from_checkpoint: null
43
+ revision: null
44
+ sample_batch_size: 4
45
+ scale_lr: false
46
+ seed: null
47
+ text_encoder_use_attention_mask: false
48
+ tokenizer_max_length: null
49
+ tokenizer_name: null
50
+ train_batch_size: 1
51
+ train_text_encoder: false
52
+ use_8bit_adam: true
53
+ validation_epochs: 50
54
+ validation_prompt: null
55
+ variant: null
56
+ with_prior_preservation: true
logs/dreambooth-lora/events.out.tfevents.1708859151.greengpu.greengpu.com.1281380.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d7e44df240fc018b3068a02da51a7ce2561c5e6e305eee147af831c858289aa
3
+ size 66774
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8fbf31a39a1d92f22e2e8ecdcd3307ff2aa3940f61c0c349fe5760dd47738cbe
3
+ size 3226184