End of training

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+samples/0000.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1 +1,44 @@
-Fine Tuned stable diffusion for tattoo designs
+
+---
+license: creativeml-openrail-m
+base_model: runwayml/stable-diffusion-v1-5
+datasets:
+- Drozdik/tattoo_v3
+tags:
+- stable-diffusion
+- stable-diffusion-diffusers
+- text-to-image
+- diffusers
+inference: true
+---
+    
+# Text-to-image finetuning - TejasNavada/tattoo-diffusion
+
+This pipeline was finetuned from **runwayml/stable-diffusion-v1-5** on the **Drozdik/tattoo_v3** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['a dragon on a white background', ' a fiery skull', 'a skull', 'a face', 'a snake and skull']: 
+
+
+
+## Pipeline usage
+
+You can use the pipeline like so:
+
+```python
+from diffusers import DiffusionPipeline
+import torch
+
+pipeline = DiffusionPipeline.from_pretrained("TejasNavada/tattoo-diffusion", torch_dtype=torch.float16)
+prompt = "a dragon on a white background"
+image = pipeline(prompt).images[0]
+image.save("my_image.png")
+```
+
+## Training info
+
+These are the key hyperparameters used during training:
+
+* Epochs: 50
+* Learning rate: 0.0001
+* Batch size: 2
+* Image resolution: 512
+* Mixed-precision: fp16
+

checkpoint-2120/optimizer.bin

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

checkpoint-2120/pytorch_model.bin

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

checkpoint-2120/random_states_0.pkl

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

checkpoint-2120/scaler.pt

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

checkpoint-2120/scheduler.bin

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

logs/train_example/events.out.tfevents.1692071459.53481a5e31cf.513.0

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