FredZhang7
commited on
Commit
•
94e9bb0
1
Parent(s):
458f530
Update README.md
Browse files
README.md
CHANGED
@@ -6,9 +6,12 @@ tags:
|
|
6 |
- distilgpt2
|
7 |
---
|
8 |
# Distilgpt2 Stable Diffusion Model Card
|
9 |
-
Distilgpt2 Stable Diffusion is a text-to-text model used to generate creative and coherent prompts given any text.
|
10 |
This model was finetuned on 2.03M stable diffusion prompts from [Stable Diffusion discord](https://huggingface.co/datasets/bartman081523/stable-diffusion-discord-prompts), [Lexica.art](https://huggingface.co/datasets/Gustavosta/Stable-Diffusion-Prompts), and (my hand-picked) [Krea.ai](./krea.ai.txt). I filtered the hand-picked prompts based on the output results from Stable Diffusion v1.4.
|
11 |
|
|
|
|
|
|
|
12 |
### PyTorch
|
13 |
|
14 |
```bash
|
|
|
6 |
- distilgpt2
|
7 |
---
|
8 |
# Distilgpt2 Stable Diffusion Model Card
|
9 |
+
Distilgpt2 Stable Diffusion is a text-to-text model used to generate creative and coherent prompts for text-to-image models, given any text.
|
10 |
This model was finetuned on 2.03M stable diffusion prompts from [Stable Diffusion discord](https://huggingface.co/datasets/bartman081523/stable-diffusion-discord-prompts), [Lexica.art](https://huggingface.co/datasets/Gustavosta/Stable-Diffusion-Prompts), and (my hand-picked) [Krea.ai](./krea.ai.txt). I filtered the hand-picked prompts based on the output results from Stable Diffusion v1.4.
|
11 |
|
12 |
+
Compared to other prompt generation models using GPT2, this one runs with 50% faster forwardpropagation and 40% less disk space & RAM.
|
13 |
+
|
14 |
+
|
15 |
### PyTorch
|
16 |
|
17 |
```bash
|