Downloads · 30 days
225
0% of all-time downloads
gpt2-650k-stable-diffusion-prompt-generator
Ar4ikov/gpt2-650k-stable-diffusion-prompt-generator
gpt2-650k-stable-diffusion-prompt-generator is a text generation model from Ar4ikov. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as mit.
All-time downloads
53.6K
Public
Parameters
137M
1 GB on disk
Likes
41
Public
Repo makeup
Click a slice to open those files.
.bin510 MB · 50%
Parameter types
How the weights are stored.
F32124M · 91%
Try a prompt
Datasets
Model card
From the Hugging Face model README
Stable Diffusion Prompt Generator
TODO: Complete me next time
Introcude
...
from transformers import pipeline
pipe = pipeline('text-generation', model_id='Ar4ikov/gpt2-650k-stable-diffusion-prompt-generator')
def get_valid_prompt(text: str) -> str:
dot_split = text.split('.')[0]
n_split = text.split('\n')[0]
return {
len(dot_split) < len(n_split): dot_split,
len(n_split) > len(dot_split): n_split,
len(n_split) == len(dot_split): dot_split
}[True]
prompt = 'A Tokio town landscape, sunset, by'
valid_prompt = get_valid_prompt(pipe(prompt, max_length=77)[0]['generated_text'])
print(valid_prompt)
# >>> A Tokio town landscape, sunset, by Greg Rutkowski,Artgerm,trending on Behance,light effect,high detail,3d sculpture,golden ratio,dramatic,dramatic background,digital art
- Task
- Text Generation
- Library
- transformers
- Type
- gpt2
- License
- mit
- Languages
- en
- Created
- Jan 11, 2023
- Updated
- Mar 22, 2023