pbevan11 commited on
Commit
52579e2
1 Parent(s): 30a9831

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -4,7 +4,11 @@ license: openrail++
4
 
5
  This is a finetuned version of [stabilityai/stable-diffusion-2-base](https://huggingface.co/stabilityai/stable-diffusion-2-base), optimised for outputting English text.
6
 
7
- The model is finetuned for 70 epochs on [pbevan11/GPT4V-captions-from-LVIS-typography](https://huggingface.co/datasets/pbevan11/GPT4V-captions-from-LVIS-typography), a curated dataset of image-caption pairs from LVIS with detailed transcriptions of the text present in the image.
 
 
 
 
8
 
9
 
10
  # Citation
 
4
 
5
  This is a finetuned version of [stabilityai/stable-diffusion-2-base](https://huggingface.co/stabilityai/stable-diffusion-2-base), optimised for outputting English text.
6
 
7
+ The model is finetuned for 20 epochs on [pbevan11/GPT4V-captions-from-LVIS-typography](https://huggingface.co/datasets/pbevan11/GPT4V-captions-from-LVIS-typography), a curated dataset of image-caption pairs from LVIS with detailed transcriptions of the text present in the image. We trained with a learning rate of 5e-6.
8
+
9
+ ---
10
+
11
+ ![Image generation model spelling comparison](model_comparison.png)
12
 
13
 
14
  # Citation