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
+val_imgs_grid.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,7 +1,7 @@
 
 ---
 license: creativeml-openrail-m
-base_model: SG161222/Realistic_Vision_V4.0
+base_model: stabilityai/stable-diffusion-2-1
 datasets:
 - fantasyfish/laion-art
 tags:
@@ -14,7 +14,7 @@ inference: true
     
 # Text-to-image finetuning - Vishnou/sd-laion-art
 
-This pipeline was finetuned from **SG161222/Realistic_Vision_V4.0** 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']: 
+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)
 
@@ -37,7 +37,7 @@ image.save("my_image.png")
 
 These are the key hyperparameters used during training:
 
-* Epochs: 2
+* Epochs: 3
 * Learning rate: 1e-05
 * Batch size: 16
 * Gradient accumulation steps: 4
@@ -45,4 +45,4 @@ These are the key hyperparameters used during training:
 * Mixed-precision: bf16
 
 
-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/nrt84fkx).
+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/l2qhp86s).