huggingface-transformers - 如何从 GPTNeo 中生成的文本中删除输入？

Question

我正在编写一个程序来生成文本......我需要从生成的文本中删除输入。我怎样才能做到这一点？编码：

input_ids = tokenizer(context, return_tensors="pt").input_ids
gen_tokens = model.generate(
    input_ids,
    do_sample=True,
    temperature=0.8,
    top_p=0.9)
strs = tokenizer.batch_decode(gen_tokens)[0]

这里的 strs 包含我给出的输入......如何删除它？