mlc-llm [Bug] gen_config 并未每次都将 context_window_size 添加到 model_config 中,

c7rzv4ha  于 2个月前  发布在  其他
关注(0)|答案(2)|浏览(37)

🐛 Bug

gen_config 并不总是将 context_window_size 添加到 model_config 中。
context_window_size 键对于 LLMEngine 的加载是必需的。
要重现此问题:在 Mistral 7B 模型上使用 gen_config,并带有 --conv-template=dolly 参数,然后尝试使用 LLMEngine() 加载该模型。
File "<snip>\.venv\Lib\site-packages\mlc_llm\serve\engine_base.py", line 258, in _get_model_config_limit compile_time_context_window_size = config["model_config"]["context_window_size"] ~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^ KeyError: 'context_window_size'
Windows + nightly wheel。已通过手动将键添加到 mlc-chat-config.json 中修复。
我知道这只是小问题,但还是想分享一下。

tzdcorbm

tzdcorbm1#

非常感谢您的挖掘和报告。我们会进行调查并修复问题!

iezvtpos

iezvtpos2#

让我们在gen_config中构建一个验证函数,以检查生成的json,并检查必需的字段。

相关问题