Views
No views yet
gpt2-parasciparaphrasegpt2)fp16 mixed precisionsentence1 → sentence2)paraphrase this sentence: {sentence1}\n{sentence2}pad_token = eos_tokenwandb)1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("your-username/gpt2-parasciparaphrase")
4tokenizer = AutoTokenizer.from_pretrained("your-username/gpt2-parasciparaphrase")
5
6input_text = "paraphrase this sentence: AI models can help in automating tasks.\n"
7input_ids = tokenizer.encode(input_text, return_tensors="pt")
8
9output = model.generate(input_ids, max_new_tokens=50, do_sample=True, top_k=50, top_p=0.95)
10print(tokenizer.decode(output[0], skip_special_tokens=True))