YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
Quantization made by Richard Erkhov.
gpt2-650k-stable-diffusion-prompt-generator - GGUF
- Model creator: https://huggingface.co/Ar4ikov/
- Original model: https://huggingface.co/Ar4ikov/gpt2-650k-stable-diffusion-prompt-generator/
Original model description:
license: mit datasets: - Gustavosta/Stable-Diffusion-Prompts - bartman081523/stable-diffusion-discord-prompts - Ar4ikov/sd_filtered_2m language: - en library_name: transformers pipeline_tag: text-generation tags: - art - code widget: - text: A Tokio town landscape, sunset, by - text: A Tokio town landscape, sunset - text: 1girl, blue eyes, dark hair - text: An astronaut, holding a wrench in outer space - text: A fire soul eater demon - text: A portrait of a beautiful woman - text: A portret of an artist man, thick beard
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
- Downloads last month
- 928
Model size
163M params
Architecture
gpt2
Unable to determine this model's library. Check the
docs
.