Views
No views yet
1model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
2
3enhancer = pipeline('text2text-generation',
4 model=model,
5 tokenizer=tokenizer,
6 repetition_penalty= 1.2,
7 device=device)
8
9max_target_length = 256
10prefix = "enhance prompt: "
11
12short_prompt = "beautiful house with text 'hello'"
13answer = enhancer(prefix + short_prompt, max_length=max_target_length)
14final_answer = answer[0]['generated_text']
15print(final_answer)