End of epoch 0

README.md

@@ -1,3 +1,4 @@
+
 ---
 license: creativeml-openrail-m
 base_model: stabilityai/stable-diffusion-2-1
@@ -15,183 +16,9 @@ inference: true
 
 This pipeline was finetuned from **stabilityai/stable-diffusion-2-1** on the **fantasyfish/laion-art** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['A man in a suit']: 
 
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
 ![val_imgs_grid](./val_imgs_grid.png)
 
 
-
 ## Pipeline usage
 
 You can use the pipeline like so:
@@ -210,7 +37,7 @@ image.save("my_image.png")
 
 These are the key hyperparameters used during training:
 
-* Epochs: 32
+* Epochs: 100
 * Learning rate: 1e-05
 * Batch size: 16
 * Gradient accumulation steps: 4
@@ -218,6 +45,4 @@ These are the key hyperparameters used during training:
 * Mixed-precision: None
 
 
-
-
-More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/vishnou/text2image-fine-tune/runs/2x8jqvkp).
+More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/vishnou/text2image-fine-tune/runs/f6tt9nwo).

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e2f4f2412bb0fb9d5916a11f29753af2a3a56140fcbc3b08f1921b1eaa9f5ed4
-size 1731904704
+oid sha256:0ec5dedec7a34dd5f1479665788360f827b44658764e91504db31875e712304d
+size 1307346120