End of training
Browse files- README.md +43 -0
- checkpoint-1600/optimizer.bin +3 -0
- checkpoint-1600/pytorch_lora_weights.safetensors +3 -0
- checkpoint-1600/pytorch_model.bin +3 -0
- checkpoint-1600/random_states_0.pkl +3 -0
- checkpoint-1600/scaler.pt +3 -0
- checkpoint-1600/scheduler.bin +3 -0
- checkpoint-800/optimizer.bin +3 -0
- checkpoint-800/pytorch_lora_weights.safetensors +3 -0
- checkpoint-800/pytorch_model.bin +3 -0
- checkpoint-800/random_states_0.pkl +3 -0
- checkpoint-800/scaler.pt +3 -0
- checkpoint-800/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/text2image-fine-tune/1711316275.5026612/events.out.tfevents.1711316275.DESKTOP-TNN0PMP.43280.1 +3 -0
- logs/text2image-fine-tune/1711316275.5046616/hparams.yml +50 -0
- logs/text2image-fine-tune/events.out.tfevents.1711316275.DESKTOP-TNN0PMP.43280.0 +3 -0
- pytorch_lora_weights.safetensors +3 -0
README.md
ADDED
@@ -0,0 +1,43 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: creativeml-openrail-m
|
3 |
+
library_name: diffusers
|
4 |
+
tags:
|
5 |
+
- stable-diffusion
|
6 |
+
- stable-diffusion-diffusers
|
7 |
+
- text-to-image
|
8 |
+
- diffusers
|
9 |
+
- diffusers-training
|
10 |
+
- lora
|
11 |
+
base_model: runwayml/stable-diffusion-v1-5
|
12 |
+
inference: true
|
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 text2image fine-tuning - iamkaikai/BAUHAUS-LORA
|
20 |
+
These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights were fine-tuned on the iamkaikai/BAUHAUS-ART dataset. You can find some example images in the following.
|
21 |
+
|
22 |
+

|
23 |
+

|
24 |
+

|
25 |
+

|
26 |
+
|
27 |
+
|
28 |
+
|
29 |
+
## Intended uses & limitations
|
30 |
+
|
31 |
+
#### How to use
|
32 |
+
|
33 |
+
```python
|
34 |
+
# TODO: add an example code snippet for running this diffusion pipeline
|
35 |
+
```
|
36 |
+
|
37 |
+
#### Limitations and bias
|
38 |
+
|
39 |
+
[TODO: provide examples of latent issues and potential remediations]
|
40 |
+
|
41 |
+
## Training details
|
42 |
+
|
43 |
+
[TODO: describe the data used to train the model]
|
checkpoint-1600/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dfe8e03b6996fcc495dab8c5b3d272e08a90e79795857428d0700f41754869e3
|
3 |
+
size 6592122
|
checkpoint-1600/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a7bec49653ca1ba8023fd7c2129e78e343a77fc0298097d795601e476d4fc457
|
3 |
+
size 3226184
|
checkpoint-1600/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bc6a103957e5c971bae088db0d5fbac3f667db43ecceb02736825ba7ef5c1c4a
|
3 |
+
size 1722724610
|
checkpoint-1600/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:75a71246d533375b8a93cf471e2ed49b37cabfa025b85819f96f1ce3672b76cf
|
3 |
+
size 14408
|
checkpoint-1600/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:727dec2e704108fe3b865a887958d8ceb10740de3cbbf987a46fb0fa239c66e2
|
3 |
+
size 988
|
checkpoint-1600/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0b4de15e9bb2faf45d96502c73ddda58428bd82dd99817b34071984205ecba6c
|
3 |
+
size 1000
|
checkpoint-800/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4fd702773a4bfd5b187d331b9065e07483dc16b9c99ab25fa1e7d75eb2d55a88
|
3 |
+
size 6592122
|
checkpoint-800/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:41767148679152d5d4cd679a97e93750d7eda715670d1718b18ad0af226da011
|
3 |
+
size 3226184
|
checkpoint-800/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:32e0b6d8cd684741778bafb20f658080f3b287c7227237549b67ab096e923d43
|
3 |
+
size 1722724610
|
checkpoint-800/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:68b039ac8e1dc5aa1ece2a9fc09f7ddadfa9fd246215d88ad20aede0ca3f3ed6
|
3 |
+
size 14408
|
checkpoint-800/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ff714d973bad5fd800a581e85354e5bcc6d536bb2686f2de9915b7bcad8c9c98
|
3 |
+
size 988
|
checkpoint-800/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e3982af4e5bd1a775ac3456371bf3d86e299587661fba76593f9aefae24d984e
|
3 |
+
size 1000
|
image_0.png
ADDED
![]() |
image_1.png
ADDED
![]() |
image_2.png
ADDED
![]() |
image_3.png
ADDED
![]() |
logs/text2image-fine-tune/1711316275.5026612/events.out.tfevents.1711316275.DESKTOP-TNN0PMP.43280.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2fa967ca5de8e234ed254049feb8fb7c628c37890f116e50ca43298a73a7eda6
|
3 |
+
size 2402
|
logs/text2image-fine-tune/1711316275.5046616/hparams.yml
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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: true
|
6 |
+
cache_dir: null
|
7 |
+
caption_column: text
|
8 |
+
center_crop: true
|
9 |
+
checkpointing_steps: 800
|
10 |
+
checkpoints_total_limit: 2
|
11 |
+
dataloader_num_workers: 0
|
12 |
+
dataset_config_name: null
|
13 |
+
dataset_name: iamkaikai/BAUHAUS-ART
|
14 |
+
enable_xformers_memory_efficient_attention: false
|
15 |
+
gradient_accumulation_steps: 1
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hub_model_id: BAUHAUS-LORA
|
18 |
+
hub_token: null
|
19 |
+
image_column: image
|
20 |
+
learning_rate: 0.0001
|
21 |
+
local_rank: -1
|
22 |
+
logging_dir: logs
|
23 |
+
lr_scheduler: constant
|
24 |
+
lr_warmup_steps: 400
|
25 |
+
max_grad_norm: 1.0
|
26 |
+
max_train_samples: null
|
27 |
+
max_train_steps: 1775
|
28 |
+
mixed_precision: fp16
|
29 |
+
noise_offset: 0
|
30 |
+
num_train_epochs: 25
|
31 |
+
num_validation_images: 4
|
32 |
+
output_dir: BAUHAUS-LORA
|
33 |
+
prediction_type: null
|
34 |
+
pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
|
35 |
+
push_to_hub: true
|
36 |
+
random_flip: true
|
37 |
+
rank: 4
|
38 |
+
report_to: tensorboard
|
39 |
+
resolution: 512
|
40 |
+
resume_from_checkpoint: null
|
41 |
+
revision: null
|
42 |
+
scale_lr: false
|
43 |
+
seed: 1337
|
44 |
+
snr_gamma: null
|
45 |
+
train_batch_size: 4
|
46 |
+
train_data_dir: null
|
47 |
+
use_8bit_adam: false
|
48 |
+
validation_epochs: 20
|
49 |
+
validation_prompt: BAUHAUS-ART, silver, brown, silver
|
50 |
+
variant: null
|
logs/text2image-fine-tune/events.out.tfevents.1711316275.DESKTOP-TNN0PMP.43280.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d6e072eb0936b37d820dfb05c7817be2ede1cd67c1b657ca9214c9ac60438839
|
3 |
+
size 2486911
|
pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1691b30d88474f776013de05dbab5f5d47a12e6777a336fc0318f68a3d254cc5
|
3 |
+
size 3226184
|