"""Prompt-template constants shared by CPU metadata builders and GPU encoding."""

# Training truncates the prompt here.  The caption prompts ask for much shorter
# targets (the length bands top out at 1,280 retained tokens), but a caption
# that came out longer than asked is kept rather than cut at the request size.
MAX_SEQUENCE_LENGTH = 2048
DROP_IDX = 38
RETAINED_MIN_LENGTH = 1
SYSTEM_PROMPT = (
    "Describe the image, focusing on its content, artistic style, composition, "
    "lighting, color, texture, and the spatial relationships between objects "
    "and the background:"
)
PROMPT_TEMPLATE = (
    "<|im_start|>system\n{system_prompt}<|im_end|>\n"
    "<|im_start|>user\n{user_prompt}<|im_end|>\n"
    "<|im_start|>assistant\n"
)