Views
No views yet
1model_name = "theblackcat102/alpaca-title-generator-mt0-large"
2tokenizer = AutoTokenizer.from_pretrained(model_name)
3model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
4question = 'Hi\nHow can I help you?'
5encodes = tokenizer(question, return_tensors='pt')
6outputs = model.generate(encodes.input_ids,
7 max_length=512,
8 do_sample=True,
9 repetition_penalty=1.2,
10 top_k=50,
11 num_return_sequences=1,
12 early_stopping=True
13)
14for i, beam_output in enumerate(outputs):
15 print('-----')
16 print("{}".format(tokenizer.decode(beam_output, skip_special_tokens=True)))
17# > Help requested.yahma/alpaca-cleaned and use openai turbo model for title.""
user: {}
assistant: {}
""
Generate a very short title within 5 words of the conversation above, title must be as relevant as possible. Title language must be same as the context
TITLE: