End of training

README.md

@@ -8,15 +8,14 @@ tags:
 - stable-diffusion-diffusers
 - text-to-image
 - diffusers
-- dreambooth
+- lora
 inference: true
 ---
     
-# DreamBooth - sotoy/path_to_saved_model
+# LoRA DreamBooth - sotoy/path_to_saved_model
 
-This is a dreambooth model derived from 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. 
+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. 
 
 
 
-DreamBooth for the text encoder was enabled: True.
+LoRA for the text encoder was enabled: True.

checkpoint-500/optimizer.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ab0311c84f76fee86c12fe48ad3752c26332bb14d6efa3a64b85aa4436249a7e
-size 1972599009
+oid sha256:1b51f0cecd1cb1f62f9d4f2f90be6ae5aecf2fcdf3117e4590e3d9694687378a
+size 6417349

checkpoint-500/pytorch_lora_weights.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c035f45d03414a8018d91a911031f6ca536bbd17f1f8125a9c827a399ad8c32c
+size 4506363

checkpoint-500/random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0725742e71ec22bb80e61d4966ec10b04995c69f52abaa462b5b7ed63bf6db79
-size 15691
+oid sha256:9ad1dfb0496b7b814012d15fe42ad3513ad05bc3f8e9816d3c2f4018e06bf005
+size 15755

logs/dreambooth-lora/1689344207.6386104/events.out.tfevents.1689344207.iti-983.37399.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:710448a14903931b8c7759612325af7cad284afdec29347fdfa85730354b1d22
+size 2654

logs/dreambooth-lora/1689344207.641934/hparams.yml

@@ -0,0 +1,56 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.01
+allow_tf32: false
+center_crop: false
+checkpointing_steps: 500
+checkpoints_total_limit: null
+class_data_dir: path_to_class_images
+class_labels_conditioning: null
+class_prompt: a photo of dog
+dataloader_num_workers: 0
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 1
+gradient_checkpointing: true
+hub_model_id: null
+hub_token: null
+instance_data_dir: ./dog
+instance_prompt: a photo of sks dog
+learning_rate: 2.0e-06
+local_rank: 0
+logging_dir: logs
+lr_num_cycles: 1
+lr_power: 1.0
+lr_scheduler: constant
+lr_warmup_steps: 0
+max_grad_norm: 1.0
+max_train_steps: 800
+mixed_precision: null
+num_class_images: 200
+num_train_epochs: 8
+num_validation_images: 4
+output_dir: path_to_saved_model
+pre_compute_text_embeddings: false
+pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
+prior_generation_precision: null
+prior_loss_weight: 1.0
+push_to_hub: true
+rank: 4
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: null
+revision: null
+sample_batch_size: 4
+scale_lr: false
+seed: null
+text_encoder_use_attention_mask: false
+tokenizer_max_length: null
+tokenizer_name: null
+train_batch_size: 1
+train_text_encoder: true
+use_8bit_adam: true
+validation_epochs: 50
+validation_images: null
+validation_prompt: null
+with_prior_preservation: true

logs/dreambooth-lora/events.out.tfevents.1689344207.iti-983.37399.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b8dfc66824bea55f77737e96dd347907da668794f3e2c3946a914588658a467c
+size 67034

pytorch_lora_weights.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da0c2e90065e0136029c427fac6785ccdd21c36eb15c0a1c5cfd94e89eb04538
+size 4506363