End of training

README.md

@@ -0,0 +1,54 @@
+
+---
+license: mit
+base_model: warp-ai/wuerstchen-prior
+datasets:
+- dongOi071102/meme-pretreatment-dataset
+tags:
+- wuerstchen
+- text-to-image
+- diffusers
+- diffusers-training
+- lora
+inference: true
+---
+    
+# LoRA Finetuning - dongOi071102/wuerstchen-prior-naruto-lora-4
+
+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']: 
+
+![val_imgs_grid](./val_imgs_grid.png)
+
+
+## Pipeline usage
+
+You can use the pipeline like so:
+
+```python
+from diffusers import DiffusionPipeline
+import torch
+
+pipeline = AutoPipelineForText2Image.from_pretrained(
+                "warp-ai/wuerstchen", torch_dtype=float32
+            )
+# load lora weights from folder:
+pipeline.prior_pipe.load_lora_weights("dongOi071102/wuerstchen-prior-naruto-lora-4", torch_dtype=float32)
+
+image = pipeline(prompt=prompt).images[0]
+image.save("my_image.png")
+```
+
+## Training info
+
+These are the key hyperparameters used during training:
+
+* LoRA rank: 4
+* Epochs: 20
+* Learning rate: 0.0001
+* Batch size: 8
+* Gradient accumulation steps: 1
+* Image resolution: 512
+* Mixed-precision: fp16
+
+
+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/1ga3ljgp).

checkpoint-1000/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:edac94059605e4e9a289e047f6c62a2645ccf6089db0bb4ea1eb43939f46c438
+size 12797562

checkpoint-1000/pytorch_lora_weights.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b7f8ebbb44f45b1eb2074f6b52cbdf1a56b39eaf274f21911a5c2ee0ac407746
+size 6320568

checkpoint-1000/random_states_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5223e82dd52dfe670cb8b6f526906e37ad0a78f9578be698f0a85cb4158c0ae
+size 14408

checkpoint-1000/scaler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1af6a0dbc2ac9987e8e4f3cefe7a93dbbdb3c5252d9e80fc105356e0a18a7c59
+size 988

checkpoint-1000/scheduler.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:27ddc4c68dd2b663be4770d007107065adcf08c8561b4bf4c438d7068bcdd3b7
+size 1000

checkpoint-2000/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cee74c6fd68c1296cd458a804aa669e8ae84fee5052d43689924a3996a9c750c
+size 12797562

checkpoint-2000/pytorch_lora_weights.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:47efbd61df140e32680a5bbcd7e915eac607cf93865fee666bd8ba0145ea6fad
+size 6320568

checkpoint-2000/random_states_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f6a50e3e7053a5fe61d25d8f91d8633b896e1b47b15ec2ed0eff0a3351997540
+size 14408

checkpoint-2000/scaler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:18955955fef08b17fc91e1f527d68c64b2c7bfff421b4d8c4998da1b4b35cbd8
+size 988

checkpoint-2000/scheduler.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a7f82f720c74cb46a582fe96b040c3f6ab5b38bf0c837db3b5d070145b7940ca
+size 1000

pytorch_lora_weights.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e9a1462fd5061f7cf6ce554f7a4451f496a0c7e76c640bfe770b2fa4dcf9f192
+size 6320568