设置max_new_tokens,就会自动截断,获取的是新的生成的文本吗?
在 Hugging Face Transformers 中,max_new_tokens
参数会限制模型生成的新 token 数量,并且返回的 test_outputs
通常只包含新生成的内容(不包括原始输入)。
一、max_new_tokens
的核心作用
1. 控制生成的新 token 数量
test_outputs = model.generate(
input_ids=test_inputs.input_ids,