rookieSJTU
commited on
Commit
•
c5c9ece
1
Parent(s):
45773b6
End of training
Browse files- README.md +47 -0
- checkpoint-1000/optimizer.bin +3 -0
- checkpoint-1000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-1000/random_states_0.pkl +3 -0
- checkpoint-1000/scheduler.bin +3 -0
- checkpoint-1500/optimizer.bin +3 -0
- checkpoint-1500/pytorch_lora_weights.safetensors +3 -0
- checkpoint-1500/random_states_0.pkl +3 -0
- checkpoint-1500/scheduler.bin +3 -0
- checkpoint-2000/optimizer.bin +3 -0
- checkpoint-2000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-2000/random_states_0.pkl +3 -0
- checkpoint-2000/scheduler.bin +3 -0
- checkpoint-2500/optimizer.bin +3 -0
- checkpoint-2500/pytorch_lora_weights.safetensors +3 -0
- checkpoint-2500/random_states_0.pkl +3 -0
- checkpoint-2500/scheduler.bin +3 -0
- checkpoint-3000/optimizer.bin +3 -0
- checkpoint-3000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-3000/random_states_0.pkl +3 -0
- checkpoint-3000/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/1721029417.401408/events.out.tfevents.1721029417.f83a41cc455f.106571.1 +3 -0
- logs/dreambooth-lora/1721029417.408552/hparams.yml +56 -0
- logs/dreambooth-lora/events.out.tfevents.1721029417.f83a41cc455f.106571.0 +3 -0
- pytorch_lora_weights.safetensors +3 -0
README.md
ADDED
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: runwayml/stable-diffusion-v1-5
|
3 |
+
library_name: diffusers
|
4 |
+
license: creativeml-openrail-m
|
5 |
+
tags:
|
6 |
+
- text-to-image
|
7 |
+
- diffusers
|
8 |
+
- lora
|
9 |
+
- diffusers-training
|
10 |
+
- stable-diffusion
|
11 |
+
- stable-diffusion-diffusers
|
12 |
+
inference: true
|
13 |
+
instance_prompt: a <zhongguoribao> line art
|
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 - rookieSJTU/zhongguoribao_model_setting2
|
21 |
+
|
22 |
+
These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights were trained on a <zhongguoribao> line art 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-1000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dcd12cb84b01c01bf441aadf5e0e9b821bdf94025ff9c46d98cb40a9a8c8cd05
|
3 |
+
size 3403386
|
checkpoint-1000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bd65d3b21b13d2aef9549261a695f3e2ac1477d373dcde79f65a2e20e949ed61
|
3 |
+
size 1631936
|
checkpoint-1000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4fdddac55c35912b4c9b7ae5b4ec1708957d00a3f101f515ae6a0d7a3501deb5
|
3 |
+
size 14540
|
checkpoint-1000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0e90f324c063fb4ded4c88988a38d8d5298ded52e612ed85b034c68c733d924e
|
3 |
+
size 1000
|
checkpoint-1500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:33400d2d7a356ee7d3896b2b1515ccd7b58e518835fc82f72bbf3dfae50d00e8
|
3 |
+
size 3403386
|
checkpoint-1500/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:269f043f5c23152c10c64fcb30166d9d6505100fd545ab73a1a68d8e377514a3
|
3 |
+
size 1631936
|
checkpoint-1500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5698d3cb471edd8639a121e17d30c78f587f64133f6ba26d25d90334b493356e
|
3 |
+
size 14540
|
checkpoint-1500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4b2518fb5a8af8c7f621d8a0f759f5e902d06a567ca32cc2c46a790646a022f2
|
3 |
+
size 1000
|
checkpoint-2000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9d5c9a6f123507788d56316d1766e4d7b7bf950e664c47c48e63704cabe085d0
|
3 |
+
size 3403386
|
checkpoint-2000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:13d3900410bf07a8e7a316e6a454b3a53a6ea370f5fdc7e1e4734dcd7308daf5
|
3 |
+
size 1631936
|
checkpoint-2000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3ec7eb368918f4a823abd6672131390eed56089946b4c481edf23c360a6b05b4
|
3 |
+
size 14540
|
checkpoint-2000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9543e96f8e6f6d9bb5e6322c497b0a06f3c120a34af9f7e377df5185601de555
|
3 |
+
size 1000
|
checkpoint-2500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e2c760f0c0a0f8e585f97e17cd2d4ad474b736b8dc4f0f3843985875cba17a1a
|
3 |
+
size 3403386
|
checkpoint-2500/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1afe16fd38ed5af4ad27c9b0ce3192b24645fddb1097adf1e93124d7db864f30
|
3 |
+
size 1631936
|
checkpoint-2500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2ac03b633dcf505b9df15da17b1340e4dfb509924503330c23a6ce0a4abacdf2
|
3 |
+
size 14540
|
checkpoint-2500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fde229e1ebbd0b741012a315a319f9b9647dcf48cc825a5f908eedce70819a8e
|
3 |
+
size 1000
|
checkpoint-3000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ed61c6d6f61ab681a251db871b51a8bf9d89aef246a6a94b55f55d5276352656
|
3 |
+
size 3403386
|
checkpoint-3000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:222c6ce523da806bafe4d0729dec025779d4e8eac26c2cfacedbb2a51d82f150
|
3 |
+
size 1631936
|
checkpoint-3000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b31ee46a95825c064c4aaf079466d18a36ef2c2304e9a95cca4f71df0b623403
|
3 |
+
size 14540
|
checkpoint-3000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a28add7eb5ea3823075fbfbf6cce836ed468b8be1dafa5c8afcc90a2fac54b6f
|
3 |
+
size 1000
|
checkpoint-500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7f2a9ad51f355aa187c7f1863f9cbebc0ed855460fbdc007a80f0086f2c539ce
|
3 |
+
size 3403386
|
checkpoint-500/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9262c348459ef28266f4b2cb2d68db97e8feffeab009e0c2addca8a7c71b1975
|
3 |
+
size 1631936
|
checkpoint-500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1ce6ae2057b547f122de58f296743138eb6650bc8210f91cc3caf113045c520e
|
3 |
+
size 14540
|
checkpoint-500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:614781bc6e22e5e65846f7005e6db20db239a55851aee9934404840242fb55b9
|
3 |
+
size 1000
|
image_0.png
ADDED
image_1.png
ADDED
image_2.png
ADDED
image_3.png
ADDED
logs/dreambooth-lora/1721029417.401408/events.out.tfevents.1721029417.f83a41cc455f.106571.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:19c1003897dbc3d6b794bdd8faf879bd174ba39570f3931d82bc5c70fe7d560e
|
3 |
+
size 2688
|
logs/dreambooth-lora/1721029417.408552/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: 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: ./zhongguoribao
|
19 |
+
instance_prompt: a <zhongguoribao> line art
|
20 |
+
learning_rate: 5.0e-06
|
21 |
+
local_rank: 0
|
22 |
+
logging_dir: logs
|
23 |
+
lr_num_cycles: 1
|
24 |
+
lr_power: 1.0
|
25 |
+
lr_scheduler: cosine_with_restarts
|
26 |
+
lr_warmup_steps: 20
|
27 |
+
max_grad_norm: 1.0
|
28 |
+
max_train_steps: 3000
|
29 |
+
mixed_precision: null
|
30 |
+
num_class_images: 100
|
31 |
+
num_train_epochs: 37
|
32 |
+
num_validation_images: 4
|
33 |
+
output_dir: zhongguoribao_model_setting2
|
34 |
+
pre_compute_text_embeddings: false
|
35 |
+
pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
|
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: 300
|
54 |
+
validation_prompt: a <zhongguoribao> style drawing
|
55 |
+
variant: null
|
56 |
+
with_prior_preservation: false
|
logs/dreambooth-lora/events.out.tfevents.1721029417.f83a41cc455f.106571.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2f7c1bfdf8bb4e089a63332e9b0c2d7ae8d25227c7555cb3791b487f121030bc
|
3 |
+
size 4866344
|
pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cc308775c03f32a6cd139ece9b56532e0410a38e1fe3c75c2521822e18279c56
|
3 |
+
size 3226184
|