End of training
Browse files- README.md +47 -0
- checkpoint-100/optimizer.bin +3 -0
- checkpoint-100/pytorch_lora_weights.safetensors +3 -0
- checkpoint-100/random_states_0.pkl +3 -0
- checkpoint-100/scheduler.bin +3 -0
- checkpoint-200/optimizer.bin +3 -0
- checkpoint-200/pytorch_lora_weights.safetensors +3 -0
- checkpoint-200/random_states_0.pkl +3 -0
- checkpoint-200/scheduler.bin +3 -0
- checkpoint-300/optimizer.bin +3 -0
- checkpoint-300/pytorch_lora_weights.safetensors +3 -0
- checkpoint-300/random_states_0.pkl +3 -0
- checkpoint-300/scheduler.bin +3 -0
- checkpoint-400/optimizer.bin +3 -0
- checkpoint-400/pytorch_lora_weights.safetensors +3 -0
- checkpoint-400/random_states_0.pkl +3 -0
- checkpoint-400/scheduler.bin +3 -0
- checkpoint-500/optimizer.bin +3 -0
- checkpoint-500/pytorch_lora_weights.safetensors +3 -0
- checkpoint-500/random_states_0.pkl +3 -0
- checkpoint-500/scheduler.bin +3 -0
- image_0.png +0 -0
- image_1.png +0 -0
- image_2.png +0 -0
- image_3.png +0 -0
- logs/dreambooth-lora/1709670228.291789/events.out.tfevents.1709670228.ae5149ec8a52.11450.1 +3 -0
- logs/dreambooth-lora/1709670228.3024516/hparams.yml +56 -0
- logs/dreambooth-lora/events.out.tfevents.1709670228.ae5149ec8a52.11450.0 +3 -0
- pytorch_lora_weights.safetensors +3 -0
README.md
ADDED
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: creativeml-openrail-m
|
3 |
+
library_name: diffusers
|
4 |
+
tags:
|
5 |
+
- text-to-image
|
6 |
+
- diffusers
|
7 |
+
- lora
|
8 |
+
- diffusers-training
|
9 |
+
- stable-diffusion
|
10 |
+
- stable-diffusion-diffusers
|
11 |
+
inference: true
|
12 |
+
base_model: CompVis/stable-diffusion-v1-4
|
13 |
+
instance_prompt: a photo of sks dog
|
14 |
+
---
|
15 |
+
|
16 |
+
<!-- This model card has been generated automatically according to the information the training script had access to. You
|
17 |
+
should probably proofread and complete it, then remove this comment. -->
|
18 |
+
|
19 |
+
|
20 |
+
# LoRA DreamBooth - BigBadWolf212/model
|
21 |
+
|
22 |
+
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.
|
23 |
+
|
24 |
+
![img_0](./image_0.png)
|
25 |
+
![img_1](./image_1.png)
|
26 |
+
![img_2](./image_2.png)
|
27 |
+
![img_3](./image_3.png)
|
28 |
+
|
29 |
+
|
30 |
+
LoRA for the text encoder was enabled: False.
|
31 |
+
|
32 |
+
|
33 |
+
## Intended uses & limitations
|
34 |
+
|
35 |
+
#### How to use
|
36 |
+
|
37 |
+
```python
|
38 |
+
# TODO: add an example code snippet for running this diffusion pipeline
|
39 |
+
```
|
40 |
+
|
41 |
+
#### Limitations and bias
|
42 |
+
|
43 |
+
[TODO: provide examples of latent issues and potential remediations]
|
44 |
+
|
45 |
+
## Training details
|
46 |
+
|
47 |
+
[TODO: describe the data used to train the model]
|
checkpoint-100/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:74acc532e5bf37ec21d8e3eb59dcb0a56921360535d1b7e7ceacb9e1d1c41331
|
3 |
+
size 6592122
|
checkpoint-100/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7eedbba9cd5a61a421dfa3ebe5fa50934c9454fecc0a0fe74faa77e366bc5c43
|
3 |
+
size 3226184
|
checkpoint-100/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:547b4952cb392d662615d8ed39f025ce108164286a339f101303f97c4c18df1b
|
3 |
+
size 14280
|
checkpoint-100/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:775169a85d4b90dd73a828433c824604dc39dbd36cf213d90c3e0c82687b3460
|
3 |
+
size 1000
|
checkpoint-200/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fed6fd8c6a6393182d5a7d89b86d83cef4058490b80ec4cf1cd5b81a754179e3
|
3 |
+
size 6592122
|
checkpoint-200/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:da53c20f9c512d15aadb34118677dfc89e06270a09018ea5133db07b7552bd88
|
3 |
+
size 3226184
|
checkpoint-200/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:aa54f5e74e376cd8618687a241701cd5597b35d648202ade2ba27fcc53c1dec1
|
3 |
+
size 14280
|
checkpoint-200/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9e440cace6589de6c72ad2cf84b033c31b67b56f5f593fdfc467c84dd99ca645
|
3 |
+
size 1000
|
checkpoint-300/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:80d29382827ce53b35d39cf3e212715c1b5318ad9e6bccdb7093792a3c9381c5
|
3 |
+
size 6592122
|
checkpoint-300/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c178d3f0517cedb0eb0d0109827e453c9fa8a37d9bade1c3c8daaf3b3c6089ed
|
3 |
+
size 3226184
|
checkpoint-300/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fdb7dd4663d7ee75a49805daa99746e5385d9776261fd5eb62cd49b9337d3509
|
3 |
+
size 14280
|
checkpoint-300/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:78866eeca7df97c7ddfcd69f7a9a96ec8f478309052c05ea1972f881237cb751
|
3 |
+
size 1000
|
checkpoint-400/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dc26aeba49384d8525aa6adfb472fd6f7b7eeb1b59f2fc4ead067aea07d8d000
|
3 |
+
size 6592122
|
checkpoint-400/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5b8d4e34660e866d2a44b37ab2809d037cb1c57d788fc2713f8496a4cffb71aa
|
3 |
+
size 3226184
|
checkpoint-400/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0cad0dccbaad87b5f47118cfbb659532c341261497b92a789d0d6a814e540238
|
3 |
+
size 14280
|
checkpoint-400/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a30fb50df034f3b2fe35529f75f39a19995d00a1e34b1527e20eb794752d2f77
|
3 |
+
size 1000
|
checkpoint-500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:eb1ab908c801e29c59377c35a0f5803b909bb7551568fe45312d852fd1752f50
|
3 |
+
size 6592122
|
checkpoint-500/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:75ceb11db09241d79de74d370b0d2b8b8d01631c20dca7e9288e03fcef721472
|
3 |
+
size 3226184
|
checkpoint-500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:54e7cb1ed8c0c9a10d258758b25cc5bbcd6a019c19159ed30b0ed17d7b8c8e60
|
3 |
+
size 14280
|
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
![]() |
image_1.png
ADDED
![]() |
image_2.png
ADDED
![]() |
image_3.png
ADDED
![]() |
logs/dreambooth-lora/1709670228.291789/events.out.tfevents.1709670228.ae5149ec8a52.11450.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:56717d0a1222844fc82c414b973c87c72ebffb7c288ce062068762a509efe041
|
3 |
+
size 2722
|
logs/dreambooth-lora/1709670228.3024516/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: 100
|
8 |
+
checkpoints_total_limit: null
|
9 |
+
class_data_dir: null
|
10 |
+
class_labels_conditioning: null
|
11 |
+
class_prompt: null
|
12 |
+
dataloader_num_workers: 0
|
13 |
+
enable_xformers_memory_efficient_attention: false
|
14 |
+
gradient_accumulation_steps: 1
|
15 |
+
gradient_checkpointing: false
|
16 |
+
hub_model_id: null
|
17 |
+
hub_token: null
|
18 |
+
instance_data_dir: /content/gdrive/MyDrive/Dreambooth/dreambooth_dataset/dog/
|
19 |
+
instance_prompt: a photo of sks dog
|
20 |
+
learning_rate: 0.0001
|
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: 500
|
29 |
+
mixed_precision: null
|
30 |
+
num_class_images: 100
|
31 |
+
num_train_epochs: 100
|
32 |
+
num_validation_images: 4
|
33 |
+
output_dir: /content/gdrive/MyDrive/Dreambooth/model/
|
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: 0
|
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: false
|
53 |
+
validation_epochs: 50
|
54 |
+
validation_prompt: A photo of sks dog in a bucket
|
55 |
+
variant: null
|
56 |
+
with_prior_preservation: false
|
logs/dreambooth-lora/events.out.tfevents.1709670228.ae5149ec8a52.11450.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c3881911d142eae5d4ceeb15575eed9e598e574043a988d3e35c8f0c08148b4a
|
3 |
+
size 4964277
|
pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:75ceb11db09241d79de74d370b0d2b8b8d01631c20dca7e9288e03fcef721472
|
3 |
+
size 3226184
|