Doesn't enhance

#1
by qpqpqpqpqpqp - opened

I asked for male creatures (not even humans) to test it, it outputs prompts for muscular women
And a problem with small LLMs like:
2+2=6?
model
Correct.

messages = [
    {"role": "system", "content": "Enhance and expand the following prompt with more details and context:"},
    {"role": "user", "content": "big male creature"}
]

prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
output = pipe(prompt, max_new_tokens=256)
print(output[0]['generated_text'])

a large, bearded male creature with a large, dark-colored head and a long, bushy-tailed neck, surrounded by a garden of flowers and foliage, the creature is centered and slightly to the left of the center of the frame, with a small background of a field with dense greenery and trees.

Sign up or log in to comment