dongOi071102 commited on
Commit
8987615
·
verified ·
1 Parent(s): a56b1b7

End of training

Browse files
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