dongOi071102
commited on
Commit
•
e46441a
1
Parent(s):
cd0119b
End of training
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- 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-10000/optimizer.bin +3 -0
- checkpoint-10000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-10000/random_states_0.pkl +3 -0
- checkpoint-10000/scaler.pt +3 -0
- checkpoint-10000/scheduler.bin +3 -0
- checkpoint-11000/optimizer.bin +3 -0
- checkpoint-11000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-11000/random_states_0.pkl +3 -0
- checkpoint-11000/scaler.pt +3 -0
- checkpoint-11000/scheduler.bin +3 -0
- checkpoint-12000/optimizer.bin +3 -0
- checkpoint-12000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-12000/random_states_0.pkl +3 -0
- checkpoint-12000/scaler.pt +3 -0
- checkpoint-12000/scheduler.bin +3 -0
- checkpoint-13000/optimizer.bin +3 -0
- checkpoint-13000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-13000/random_states_0.pkl +3 -0
- checkpoint-13000/scaler.pt +3 -0
- checkpoint-13000/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/scaler.pt +3 -0
- checkpoint-2000/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/scaler.pt +3 -0
- checkpoint-3000/scheduler.bin +3 -0
- checkpoint-4000/optimizer.bin +3 -0
- checkpoint-4000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-4000/random_states_0.pkl +3 -0
- checkpoint-4000/scaler.pt +3 -0
- checkpoint-4000/scheduler.bin +3 -0
- checkpoint-5000/optimizer.bin +3 -0
- checkpoint-5000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-5000/random_states_0.pkl +3 -0
- checkpoint-5000/scaler.pt +3 -0
- checkpoint-5000/scheduler.bin +3 -0
- checkpoint-6000/optimizer.bin +3 -0
- checkpoint-6000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-6000/random_states_0.pkl +3 -0
- checkpoint-6000/scaler.pt +3 -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-4
|
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-4", 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: 100
|
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/xmsqasc9).
|
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-10000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8dc8839f50f8f57f88ffb58c4f737530e6d5285802cd17bd80559473324f5ec3
|
3 |
+
size 12797562
|
checkpoint-10000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cd1f1822b6b1efb9f4611b5c09f751f45ea7b09cf3ad3b2ecc33100637229407
|
3 |
+
size 6320568
|
checkpoint-10000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b88521b8645cb6e9c99aa8c1c323fb980092d42b00452f7e68c58fa9a1309185
|
3 |
+
size 14408
|
checkpoint-10000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ac95669cbe7843f5463b6d54bac1ee21aae0ad7e099eb6115508fcf037e2d1bb
|
3 |
+
size 988
|
checkpoint-10000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f70a92d7641b91d4bb7c58e1ceff803ba5a7f525c3e9f76798fa778db1a325d7
|
3 |
+
size 1000
|
checkpoint-11000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a9b8caf532f7cc111961921681d7724e2b8b4207e9cd3b42d05f5d72767551dd
|
3 |
+
size 12797562
|
checkpoint-11000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f8d410884d9b8904b6af15514f1c0e4e40f2588e356f8cb3872165492c6d1f92
|
3 |
+
size 6320568
|
checkpoint-11000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:46a0a978d7d1d3b9019db3ab23331ab56e3abfb5d49b3d52682088cfa3d13805
|
3 |
+
size 14408
|
checkpoint-11000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:923ef539cd97b167fe43234c7639245eb6ad45d3451a6a4eed301da1f7237b6c
|
3 |
+
size 988
|
checkpoint-11000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2edae02cb76a9c7b0f2ef562e7990dc3bd90dbfe9fbe23b48274fafcb68004fe
|
3 |
+
size 1000
|
checkpoint-12000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f91419221d711fb6c37d4bdcc642586bdf82778b224468f4cddb76a969f928c7
|
3 |
+
size 12797562
|
checkpoint-12000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3d2dc7efba4972abf7d1ce3e75ded9c911d9101c0f948d6fa4f5e60d3483bb90
|
3 |
+
size 6320568
|
checkpoint-12000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b60a394769771307ad2865ac9880b0b5171cd275649026b4ba77f95ee0fa6b5d
|
3 |
+
size 14408
|
checkpoint-12000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6e1d50254f7978442c3f24762e85854dbfc53ce515b1df9f64e23666153a3d03
|
3 |
+
size 988
|
checkpoint-12000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:64b79e01a971ed8011dd32eee6758e1557fdb7a437b7bf145a4a6d5af29c7c27
|
3 |
+
size 1000
|
checkpoint-13000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c9df91b1f8791ea4512f93f83b8477d613aaee9f8c347130fa6b292b74c0079f
|
3 |
+
size 12797562
|
checkpoint-13000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6a278faa92022091d337b076dcfb3e5bc445901d73bb70f325391928fb56fa8c
|
3 |
+
size 6320568
|
checkpoint-13000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:31b1b29f0800306da61ab4ed7a81c0ce4328532a5ff78924778a3fbb9aeff607
|
3 |
+
size 14408
|
checkpoint-13000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:50810e004bac6481cfd2373f58cbaed4b52c413b103608feed2c3e12ec827e8f
|
3 |
+
size 988
|
checkpoint-13000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3fce184c9f5b259b8fea009e6ef4ea152048cce8ba4b6be0332fba0d5e75f343
|
3 |
+
size 1000
|
checkpoint-2000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cee74c6fd68c1296cd458a804aa669e8ae84fee5052d43689924a3996a9c750c
|
3 |
+
size 12797562
|
checkpoint-2000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:47efbd61df140e32680a5bbcd7e915eac607cf93865fee666bd8ba0145ea6fad
|
3 |
+
size 6320568
|
checkpoint-2000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f6a50e3e7053a5fe61d25d8f91d8633b896e1b47b15ec2ed0eff0a3351997540
|
3 |
+
size 14408
|
checkpoint-2000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:18955955fef08b17fc91e1f527d68c64b2c7bfff421b4d8c4998da1b4b35cbd8
|
3 |
+
size 988
|
checkpoint-2000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a7f82f720c74cb46a582fe96b040c3f6ab5b38bf0c837db3b5d070145b7940ca
|
3 |
+
size 1000
|
checkpoint-3000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fe81aac73fe5a009568183712b4b9da23ce10fed41e8dad07860c949cb9b4a99
|
3 |
+
size 12797562
|
checkpoint-3000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2028527c66646f81a2c230e0f4521030bdf24561d38dac78816169432122f58d
|
3 |
+
size 6320568
|
checkpoint-3000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d5be44242e29c2f8c5f7623af7633f3f8d2323ad975f4f93025b2518ff8f5ea2
|
3 |
+
size 14408
|
checkpoint-3000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:701ae9eaebaa3e9c7d474e69045ceee60387ace292dd84f8535aba2fe99f08e6
|
3 |
+
size 988
|
checkpoint-3000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3ec4a7cfe14a5940fe76c40ec22ec149da4c207b589bed9e3dca5c07c6d194c6
|
3 |
+
size 1000
|
checkpoint-4000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6a63a4ed2778f32ccedb04e555d1e6921a5ee98ef20488cc0246ac71f51c6f0d
|
3 |
+
size 12797562
|
checkpoint-4000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:eb16eaa44307e46b1d6fe999709b6fc60bd731cc03b8300c6f5e7d590f83ea43
|
3 |
+
size 6320568
|
checkpoint-4000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5a2fd57d30012ac367aeaf2fecb88bdb1c8bbee92493fc6fbd24e932a61c9faf
|
3 |
+
size 14408
|
checkpoint-4000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c0049c72dcd7a8e2c14ea90e9d61e77261b466ab0fcbbefd3b566dacb9a11981
|
3 |
+
size 988
|
checkpoint-4000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ebcfae2c2b570f150e477c452f9c35da780bbf43850aaf14572a9abd51cb6f92
|
3 |
+
size 1000
|
checkpoint-5000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:37ec6e521728ffc85d17591d7950529cdc5543ca83143d6edba1094f967c81ca
|
3 |
+
size 12797562
|
checkpoint-5000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:df696eae705a57fcdf7fd0200943255e49bc2521ce701288d9f6a517ab51f463
|
3 |
+
size 6320568
|
checkpoint-5000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:eec5a7e389f073c098d537d87538742177fac4d42cf463210cb9175ecf958b94
|
3 |
+
size 14408
|
checkpoint-5000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ec36c2172aae90a1d5fcfabe943f4ff89fc82879610a921796d6e4f025b20df3
|
3 |
+
size 988
|
checkpoint-5000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e18e663c90abe1988a2a603d9806e15c4f234871ab5d0d437ff72c9a535bde2c
|
3 |
+
size 1000
|
checkpoint-6000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f57ad40e841010c679b6731fbad6a187cc8dcea51d06284b066ebb0293dffd0b
|
3 |
+
size 12797562
|
checkpoint-6000/pytorch_lora_weights.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:09ede90224615b0931914cfa5b1a976b42b85c3354b7e831101f796814ae923b
|
3 |
+
size 6320568
|
checkpoint-6000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a414b385ca43bac0436a7f7af26336546db3c329bb83b9a44559a2a91d791cc7
|
3 |
+
size 14408
|
checkpoint-6000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e98e36d6b2956f0d5459f46c1a704ba2e7433740227383d35e48210ef8182c38
|
3 |
+
size 988
|