ntk上下文扩展

[tkone2018]

提交前必须检查以下项目

• [X] 请确保使用的是仓库最新代码（git pull），一些问题已被解决和修复。
• [X] 我已阅读项目文档和FAQ章节并且已在Issue中对问题进行了搜索，没有找到相似问题和解决方案。
• [X] 第三方插件问题：例如llama.cpp、LangChain、text-generation-webui等，同时建议到对应的项目中查找解决方案。

问题类型

模型推理

基础模型

Chinese-Alpaca-2 (7B/13B)

操作系统

Linux

详细描述问题

generation_config = GenerationConfig( temperature=0.2, top_k=40, top_p=0.9, do_sample=True, num_beams=1, repetition_penalty=1.1, max_new_tokens=30000 ) 推理文件是inference_hf.py，采用ntk扩展，想问下是不是输入还是限制在4096了，请问下ntk是可以扩展输入的上下文吗，还是说我使用ntk后，还需要修改哪些部分才能扩展输入的上下文token数，谢谢

依赖情况（代码类问题务必提供）

# 请在此处粘贴依赖情况（请粘贴在本代码块里）

运行日志或截图

# 请在此处粘贴运行日志（请粘贴在本代码块里）

[tkone2018]

还有就是请教下vllm推理是不是不支持ntk上下文扩展技术

[iMountTai]

你max_new_token设置的太大，参数不对。而且NTK不是无限扩展，比如原来支持4k长度，NTK可扩展到8-12k。

[tkone2018]

@iMountTai 好的，谢谢

[github-actions[bot]]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your consideration.

[github-actions[bot]]

Closing the issue, since no updates observed. Feel free to re-open if you need any further assistance.