dongOi071102
commited on
End of training
Browse files- README.md +54 -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/scaler.pt +3 -0
- checkpoint-1000/scheduler.bin +3 -0
- checkpoint-1200/optimizer.bin +3 -0
- checkpoint-1200/pytorch_lora_weights.safetensors +3 -0
- checkpoint-1200/random_states_0.pkl +3 -0
- checkpoint-1200/scaler.pt +3 -0
- checkpoint-1200/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/scaler.pt +3 -0
- checkpoint-200/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/scaler.pt +3 -0
- checkpoint-400/scheduler.bin +3 -0
- checkpoint-600/optimizer.bin +3 -0
- checkpoint-600/pytorch_lora_weights.safetensors +3 -0
- checkpoint-600/random_states_0.pkl +3 -0
- checkpoint-600/scaler.pt +3 -0
- checkpoint-600/scheduler.bin +3 -0
- checkpoint-800/optimizer.bin +3 -0
- checkpoint-800/pytorch_lora_weights.safetensors +3 -0
- checkpoint-800/random_states_0.pkl +3 -0
- checkpoint-800/scaler.pt +3 -0
- checkpoint-800/scheduler.bin +3 -0
- pytorch_lora_weights.safetensors +3 -0
- val_imgs_grid.png +0 -0
README.md
ADDED
@@ -0,0 +1,54 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
---
|
3 |
+
license: mit
|
4 |
+
base_model: warp-ai/wuerstchen-prior
|
5 |
+
datasets:
|
6 |
+
- dongOi071102/meme-pretreatment-dataset
|
7 |
+
tags:
|
8 |
+
- wuerstchen
|
9 |
+
- text-to-image
|
10 |
+
- diffusers
|
11 |
+
- diffusers-training
|
12 |
+
- lora
|
13 |
+
inference: true
|
14 |
+
---
|
15 |
+
|
16 |
+
# LoRA Finetuning - dongOi071102/wuerstchen-prior-meme-lora-epochs-10
|
17 |
+
|
18 |
+
This pipeline was finetuned from **warp-ai/wuerstchen-prior** on the **dongOi071102/meme-pretreatment-dataset** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['a catton cat with angry face']:
|
19 |
+
|
20 |
+
![val_imgs_grid](./val_imgs_grid.png)
|
21 |
+
|
22 |
+
|
23 |
+
## Pipeline usage
|
24 |
+
|
25 |
+
You can use the pipeline like so:
|
26 |
+
|
27 |
+
```python
|
28 |
+
from diffusers import DiffusionPipeline
|
29 |
+
import torch
|
30 |
+
|
31 |
+
pipeline = AutoPipelineForText2Image.from_pretrained(
|
32 |
+
"warp-ai/wuerstchen", torch_dtype=float32
|
33 |
+
)
|
34 |
+
# load lora weights from folder:
|
35 |
+
pipeline.prior_pipe.load_lora_weights("dongOi071102/wuerstchen-prior-meme-lora-epochs-10", torch_dtype=float32)
|
36 |
+
|
37 |
+
image = pipeline(prompt=prompt).images[0]
|
38 |
+
image.save("my_image.png")
|
39 |
+
```
|
40 |
+
|
41 |
+
## Training info
|
42 |
+
|
43 |
+
These are the key hyperparameters used during training:
|
44 |
+
|
45 |
+
* LoRA rank: 4
|
46 |
+
* Epochs: 10
|
47 |
+
* Learning rate: 0.0001
|
48 |
+
* Batch size: 8
|
49 |
+
* Gradient accumulation steps: 1
|
50 |
+
* Image resolution: 512
|
51 |
+
* Mixed-precision: fp16
|
52 |
+
|
53 |
+
|
54 |
+
More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/2111818-no/text2image-fine-tune/runs/64ixewdo).
|
checkpoint-1000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:edac94059605e4e9a289e047f6c62a2645ccf6089db0bb4ea1eb43939f46c438
|
3 |
+
size 12797562
|
checkpoint-1000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b7f8ebbb44f45b1eb2074f6b52cbdf1a56b39eaf274f21911a5c2ee0ac407746
|
3 |
+
size 6320568
|
checkpoint-1000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a5223e82dd52dfe670cb8b6f526906e37ad0a78f9578be698f0a85cb4158c0ae
|
3 |
+
size 14408
|
checkpoint-1000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1af6a0dbc2ac9987e8e4f3cefe7a93dbbdb3c5252d9e80fc105356e0a18a7c59
|
3 |
+
size 988
|
checkpoint-1000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:27ddc4c68dd2b663be4770d007107065adcf08c8561b4bf4c438d7068bcdd3b7
|
3 |
+
size 1000
|
checkpoint-1200/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:74db80fe8aa2cc17ef00cc6530732328502f9ded7f29489df767cbe80d386eef
|
3 |
+
size 12797562
|
checkpoint-1200/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:379d24706f090ed6a6a1cea8710b8508c423441a5bebef4c5d79f2f198cde360
|
3 |
+
size 6320568
|
checkpoint-1200/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2bc3f64d73258878bfc4b5335704ddcd66a95538a55c537532968c2d35256b61
|
3 |
+
size 14408
|
checkpoint-1200/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:40e597489d1cd8a95d3d4043dc0993d67e0f34bd5b423216f8bccd4129a2f4e6
|
3 |
+
size 988
|
checkpoint-1200/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:de55d3014a1a9bd689e3f66dcca644a0a4ef5e0049ff22d906a1992d27b407d4
|
3 |
+
size 1000
|
checkpoint-200/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:de3764b6550ae01d80b55b46709bb84223992596a2b0d315b183f52271e41948
|
3 |
+
size 12797562
|
checkpoint-200/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4ee3d28c204a247e758081bf6097fe6f2b4062cb5ed326d118e1effdfe026363
|
3 |
+
size 6320568
|
checkpoint-200/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6b50091acbd45c22aa058630f43912f7c7618e152a12b1aac725d32e8f921e7c
|
3 |
+
size 14408
|
checkpoint-200/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0a684c9cd541decfaa9fe2ad786dfb7cfdc1f5a180370c6ff4f45bd2d354f470
|
3 |
+
size 988
|
checkpoint-200/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b2582099ec8761005eacab8b344959ff481ed2bd8e711a7e8256769bd5a212ee
|
3 |
+
size 1000
|
checkpoint-400/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:aeab91ee059956f39fd634dbe0b964b05fda576c5c7fc82c88663726747bf884
|
3 |
+
size 12797562
|
checkpoint-400/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:19cee32e3b1f3e9542f276672ac895857d67f9fa980568cc2486aeb4249a16d8
|
3 |
+
size 6320568
|
checkpoint-400/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:787b8edcae2b401e378baa7c1891018c3de8c1cf980194a9e70d96234b0500bc
|
3 |
+
size 14408
|
checkpoint-400/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d3ec48fb6e6a5ae6882a8f2a91c0b28cd33471de23467ad1193fa4b7708dbb97
|
3 |
+
size 988
|
checkpoint-400/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c04ac1c557aa9aaf1103361fb069ab788110426f1433683add0080f5c2c932aa
|
3 |
+
size 1000
|
checkpoint-600/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:60c29dc6674e1ddc7ad6064733bfed9853f0dba2f723ec766f2d91ac012b982d
|
3 |
+
size 12797562
|
checkpoint-600/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3dcfd87c1dac0e1f6f7397642c25d4a51e5dc9c4cc3b7fa2312e14324fcb9f2e
|
3 |
+
size 6320568
|
checkpoint-600/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ce8cf900a802443ee0503f156cd6ac4bcbc10f66a9eb16d07cff21f0f82e41f8
|
3 |
+
size 14408
|
checkpoint-600/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5b023cfc785271415c569a715e8c2a3394bab2e98ab7d32cfaa45d17611fb46a
|
3 |
+
size 988
|
checkpoint-600/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7b1c1c9a94e87808349833e0ab1df62d920caf7a00a8485989c34cba19ddd692
|
3 |
+
size 1000
|
checkpoint-800/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:09c3e0368633abd64f164f719df460667cf9bf6f4a4b7f48f3303547e9fe7ad9
|
3 |
+
size 12797562
|
checkpoint-800/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ec8cb6df27d31e4c7973d8a2e8ee5ed3a34bd9ac3018484a3a8bbdffaabd98b6
|
3 |
+
size 6320568
|
checkpoint-800/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f9e4335eccda8db9a745f5ea41ddfde19766f2cc3d8da442710471f70afbd285
|
3 |
+
size 14408
|
checkpoint-800/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6c49498b492dd47d6bea9daf708ac2579dc8325bc89ced341126fc87b06897aa
|
3 |
+
size 988
|
checkpoint-800/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7ac6b974d454808bb40e445b4ecfc5d0a6463b2a53284ae9a039b90349c862ab
|
3 |
+
size 1000
|
pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:643193481551b4b8e92a165d596f2700a43dc2e0e32ed96e96161b56f5bdc730
|
3 |
+
size 6320568
|
val_imgs_grid.png
ADDED