trollek's picture
Update README.md
f6d0415 verified
metadata
license: apache-2.0
datasets:
  - Gustavosta/Stable-Diffusion-Prompts
  - trollek/Mouse-Diffusion-Instruct
base_model: Qwen/Qwen2-0.5B
tags:
  - llama-factory
  - unsloth
language:
  - en

Qwen2-0.5B-DiffusionPrompter-v0.1

A small SD prompt model. Give it a prompt like "a beach ball" and you'll get something like:

A group of children play on a beach, wearing colorful clothing and hats. The sun is setting, casting a warm orange glow over the scene. The air is filled with laughter and screams as they jump and roll around in the sand. A small beach ball floats in the water, adding to the serene ambiance of the scene.

or "painting of a beach ball, WLOP":

beautiful sunset on the ocean in the style of wlop, painting of a beach ball, detailed and intricate details, realistic

Template

<|im_start|>user
{{image_concept}}<|im_end|>
<|im_start|>assistant
{{expanded_sd_response}}<|im_end|>

Quants and Ollama

GGUF

Ollama

ollama pull trollek/qwen2-diffusion-prompter:v01-q4_K_S
ollama pull trollek/qwen2-diffusion-prompter:v01-q5_K_S
ollama pull trollek/qwen2-diffusion-prompter:v01-q6_K

Notes and gratitude

Thanks to roborovski for superprompt-v1 and Gustavosta for Stable-Diffusion-Prompts.

I have no idea why the parameter count has grown be 100M parameters.