ihgn/

gpt2-paraphrase

  outputs = model.generate(
      input_ids, temperature=temperature, repetition_penalty=repetition_penalty,
      num_return_sequences=num_return_sequences, no_repeat_ngram_size=no_repeat_ngram_size,
      num_beams=num_beams, num_beam_groups=num_beam_groups,
      max_length=max_length, diversity_penalty=diversity_penalty
  )

  res = tokenizer.batch_decode(outputs, skip_special_tokens=True)

  return res