p1atdev's picture
Update README.md
80b8bb0 verified
metadata
library_name: transformers
license: cc-by-sa-4.0
base_model:
  - p1atdev/dart-v3-pretrain-preview-E
- name: v3 sft preview E
  version: v3
  model_name_or_path: p1atdev/dart-v3-sft-preview-E
  model_type: eager
  prompt_template_id: V3_SFT

prompt format:

RATING = Literal["<|rating:general|>", "<|rating:sensitive|>", "<|rating:questionable|>", <|rating:explicit|>]
ASPECT_RATIO = Literal[
  "<|aspect_ratio:too_tall|>",
  "<|aspect_ratio:tall_wallpaper|>",
  "<|aspect_ratio:tall|>",
  "<|aspect_ratio:square|>",
  "<|aspect_ratio:wide|>",
  "<|aspect_ratio:wide_wallpaper|>", 
  "<|aspect_ratio:too_tall|>"
]
LENGTH = Literal["<|length:very_short|>", "<|length:medium|>", "<|length:very_long|>"]

template = (
  "<|bos|>"
  "{rating}{aspect_ratio}{length}"
  "<copyright>{copyright}</copyright>"
  "<character>{character}</character>"
  "<general>{condition}<|input_end|>"
)

# for example:
prompt = template.format(
  rating="<|rating:general|>",  # RATING
  aspect_ratio="<|aspect_ratio:tall|>",  # ASPECT_RATIO
  length="<|length:medium|>",  # LENGTH
  copyright="original",
  character="my character",
  condition="1girl, solo, full body", 
)