Update README.md
Browse files
README.md
CHANGED
|
@@ -7,7 +7,14 @@ model-index:
|
|
| 7 |
---
|
| 8 |
# aksty/promptgen: Prompt generation for Text-to-Image Models
|
| 9 |
|
|
|
|
|
|
|
| 10 |
This is a text generation model trained on data specifically designed to generate prompts for text-to-image models. It is based on the [EleutherAI/gpt-neo-125M](https://huggingface.co/EleutherAI/gpt-neo-125M) pre-trained model, which has been fine-tuned using the [Gustavosta/Stable-Diffusion-Prompts](https://huggingface.co/datasets/Gustavosta/Stable-Diffusion-Prompts) dataset.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
|
| 12 |
|
| 13 |
|
|
|
|
| 7 |
---
|
| 8 |
# aksty/promptgen: Prompt generation for Text-to-Image Models
|
| 9 |
|
| 10 |
+
|
| 11 |
+
|
| 12 |
This is a text generation model trained on data specifically designed to generate prompts for text-to-image models. It is based on the [EleutherAI/gpt-neo-125M](https://huggingface.co/EleutherAI/gpt-neo-125M) pre-trained model, which has been fine-tuned using the [Gustavosta/Stable-Diffusion-Prompts](https://huggingface.co/datasets/Gustavosta/Stable-Diffusion-Prompts) dataset.
|
| 13 |
+
|
| 14 |
+
### Notebook with promptgen + Stable Diffusion v2
|
| 15 |
+
[](https://colab.research.google.com/github/aksty/promptgen/blob/main/notebook/StableDiffusion_with_PromptGen.ipynb)
|
| 16 |
+
|
| 17 |
+
|
| 18 |
|
| 19 |
|
| 20 |
|