mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-11-05 01:07:28 +00:00
[GH-ISSUE #616] 使用最新版本0.2.1,需要flash-attention的什么版本呢?是否还支持internlm2_5-7b-chat-1m模型? #456
Labels
No milestone
No project
No assignees
1 participant
Notifications
Due date
No due date set.
Dependencies
No dependencies set.
Reference
kvcache-ai/ktransformers#456
Loading…
Add table
Add a link
Reference in a new issue
No description provided.
Delete branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
Originally created by @4thGardenOfQMH on GitHub (Feb 23, 2025).
Original GitHub issue: https://github.com/kvcache-ai/ktransformers/issues/616
Originally assigned to: @Azure-Tang, @ovowei, @Atream on GitHub.
我试图使用命令 python ./ktransformers/local_chat.py --model_path /host/internlm2_5-7b-chat-1m --gguf_path /host/internlm2_5-7b-chat-1m-gguf --cpu_infer 33 --max_new_tokens 1000 --mode "/***/ktransformers-0.2.1/ktransformers/optimize/optimize_rules/Internlm2_5-7b-Chat-1m.yaml"启动internlm2_5-7b-chat-1m模型,使用internlm2_5-7b-chat-1m-q4_k_m.gguf,下载的是上海人工智能实验室提供的量化后版本,但是遇到了如下报错,请问是模型有问题,还是不支持呢?如果还需要什么信息,请告诉我,多谢!
`
You are attempting to use Flash Attention 2.0 without specifying a torch dtype. This might lead to unexpected behaviour
please input the path of your rule file(yaml file containing optimize rules):/host/ktransformers-0.2.1/ktransformers/optimize/optimize_rules/Internlm2_5-7b-Chat-1m.yaml
Injecting model as default
Injecting model.tok_embeddings as default
Injecting model.layers as default
Injecting model.layers.0 as default
Injecting model.layers.0.attention as default
Injecting model.layers.0.attention.wqkv as default
Injecting model.layers.0.attention.wo as default
Injecting model.layers.0.attention.rotary_emb as default
Injecting model.layers.0.feed_forward as default
Injecting model.layers.0.feed_forward.w1 as default
Injecting model.layers.0.feed_forward.w3 as default
Injecting model.layers.0.feed_forward.w2 as default
Injecting model.layers.0.feed_forward.act_fn as default
Injecting model.layers.0.attention_norm as default
Injecting model.layers.0.ffn_norm as default
Injecting model.layers.1 as default
Injecting model.layers.1.attention as default
Injecting model.layers.1.attention.wqkv as default
Injecting model.layers.1.attention.wo as default
Injecting model.layers.1.attention.rotary_emb as default
Injecting model.layers.1.feed_forward as default
Injecting model.layers.1.feed_forward.w1 as default
Injecting model.layers.1.feed_forward.w3 as default
Injecting model.layers.1.feed_forward.w2 as default
Injecting model.layers.1.feed_forward.act_fn as default
Injecting model.layers.1.attention_norm as default
Injecting model.layers.1.ffn_norm as default
Injecting model.layers.2 as default
Injecting model.layers.2.attention as default
Injecting model.layers.2.attention.wqkv as default
Injecting model.layers.2.attention.wo as default
Injecting model.layers.2.attention.rotary_emb as default
Injecting model.layers.2.feed_forward as default
Injecting model.layers.2.feed_forward.w1 as default
Injecting model.layers.2.feed_forward.w3 as default
Injecting model.layers.2.feed_forward.w2 as default
Injecting model.layers.2.feed_forward.act_fn as default
Injecting model.layers.2.attention_norm as default
Injecting model.layers.2.ffn_norm as default
Injecting model.layers.3 as default
Injecting model.layers.3.attention as default
Injecting model.layers.3.attention.wqkv as default
Injecting model.layers.3.attention.wo as default
Injecting model.layers.3.attention.rotary_emb as default
Injecting model.layers.3.feed_forward as default
Injecting model.layers.3.feed_forward.w1 as default
Injecting model.layers.3.feed_forward.w3 as default
Injecting model.layers.3.feed_forward.w2 as default
Injecting model.layers.3.feed_forward.act_fn as default
Injecting model.layers.3.attention_norm as default
Injecting model.layers.3.ffn_norm as default
Injecting model.layers.4 as default
Injecting model.layers.4.attention as default
Injecting model.layers.4.attention.wqkv as default
Injecting model.layers.4.attention.wo as default
Injecting model.layers.4.attention.rotary_emb as default
Injecting model.layers.4.feed_forward as default
Injecting model.layers.4.feed_forward.w1 as default
Injecting model.layers.4.feed_forward.w3 as default
Injecting model.layers.4.feed_forward.w2 as default
Injecting model.layers.4.feed_forward.act_fn as default
Injecting model.layers.4.attention_norm as default
Injecting model.layers.4.ffn_norm as default
Injecting model.layers.5 as default
Injecting model.layers.5.attention as default
Injecting model.layers.5.attention.wqkv as default
Injecting model.layers.5.attention.wo as default
Injecting model.layers.5.attention.rotary_emb as default
Injecting model.layers.5.feed_forward as default
Injecting model.layers.5.feed_forward.w1 as default
Injecting model.layers.5.feed_forward.w3 as default
Injecting model.layers.5.feed_forward.w2 as default
Injecting model.layers.5.feed_forward.act_fn as default
Injecting model.layers.5.attention_norm as default
Injecting model.layers.5.ffn_norm as default
Injecting model.layers.6 as default
Injecting model.layers.6.attention as default
Injecting model.layers.6.attention.wqkv as default
Injecting model.layers.6.attention.wo as default
Injecting model.layers.6.attention.rotary_emb as default
Injecting model.layers.6.feed_forward as default
Injecting model.layers.6.feed_forward.w1 as default
Injecting model.layers.6.feed_forward.w3 as default
Injecting model.layers.6.feed_forward.w2 as default
Injecting model.layers.6.feed_forward.act_fn as default
Injecting model.layers.6.attention_norm as default
Injecting model.layers.6.ffn_norm as default
Injecting model.layers.7 as default
Injecting model.layers.7.attention as default
Injecting model.layers.7.attention.wqkv as default
Injecting model.layers.7.attention.wo as default
Injecting model.layers.7.attention.rotary_emb as default
Injecting model.layers.7.feed_forward as default
Injecting model.layers.7.feed_forward.w1 as default
Injecting model.layers.7.feed_forward.w3 as default
Injecting model.layers.7.feed_forward.w2 as default
Injecting model.layers.7.feed_forward.act_fn as default
Injecting model.layers.7.attention_norm as default
Injecting model.layers.7.ffn_norm as default
Injecting model.layers.8 as default
Injecting model.layers.8.attention as default
Injecting model.layers.8.attention.wqkv as default
Injecting model.layers.8.attention.wo as default
Injecting model.layers.8.attention.rotary_emb as default
Injecting model.layers.8.feed_forward as default
Injecting model.layers.8.feed_forward.w1 as default
Injecting model.layers.8.feed_forward.w3 as default
Injecting model.layers.8.feed_forward.w2 as default
Injecting model.layers.8.feed_forward.act_fn as default
Injecting model.layers.8.attention_norm as default
Injecting model.layers.8.ffn_norm as default
Injecting model.layers.9 as default
Injecting model.layers.9.attention as default
Injecting model.layers.9.attention.wqkv as default
Injecting model.layers.9.attention.wo as default
Injecting model.layers.9.attention.rotary_emb as default
Injecting model.layers.9.feed_forward as default
Injecting model.layers.9.feed_forward.w1 as default
Injecting model.layers.9.feed_forward.w3 as default
Injecting model.layers.9.feed_forward.w2 as default
Injecting model.layers.9.feed_forward.act_fn as default
Injecting model.layers.9.attention_norm as default
Injecting model.layers.9.ffn_norm as default
Injecting model.layers.10 as default
Injecting model.layers.10.attention as default
Injecting model.layers.10.attention.wqkv as default
Injecting model.layers.10.attention.wo as default
Injecting model.layers.10.attention.rotary_emb as default
Injecting model.layers.10.feed_forward as default
Injecting model.layers.10.feed_forward.w1 as default
Injecting model.layers.10.feed_forward.w3 as default
Injecting model.layers.10.feed_forward.w2 as default
Injecting model.layers.10.feed_forward.act_fn as default
Injecting model.layers.10.attention_norm as default
Injecting model.layers.10.ffn_norm as default
Injecting model.layers.11 as default
Injecting model.layers.11.attention as default
Injecting model.layers.11.attention.wqkv as default
Injecting model.layers.11.attention.wo as default
Injecting model.layers.11.attention.rotary_emb as default
Injecting model.layers.11.feed_forward as default
Injecting model.layers.11.feed_forward.w1 as default
Injecting model.layers.11.feed_forward.w3 as default
Injecting model.layers.11.feed_forward.w2 as default
Injecting model.layers.11.feed_forward.act_fn as default
Injecting model.layers.11.attention_norm as default
Injecting model.layers.11.ffn_norm as default
Injecting model.layers.12 as default
Injecting model.layers.12.attention as default
Injecting model.layers.12.attention.wqkv as default
Injecting model.layers.12.attention.wo as default
Injecting model.layers.12.attention.rotary_emb as default
Injecting model.layers.12.feed_forward as default
Injecting model.layers.12.feed_forward.w1 as default
Injecting model.layers.12.feed_forward.w3 as default
Injecting model.layers.12.feed_forward.w2 as default
Injecting model.layers.12.feed_forward.act_fn as default
Injecting model.layers.12.attention_norm as default
Injecting model.layers.12.ffn_norm as default
Injecting model.layers.13 as default
Injecting model.layers.13.attention as default
Injecting model.layers.13.attention.wqkv as default
Injecting model.layers.13.attention.wo as default
Injecting model.layers.13.attention.rotary_emb as default
Injecting model.layers.13.feed_forward as default
Injecting model.layers.13.feed_forward.w1 as default
Injecting model.layers.13.feed_forward.w3 as default
Injecting model.layers.13.feed_forward.w2 as default
Injecting model.layers.13.feed_forward.act_fn as default
Injecting model.layers.13.attention_norm as default
Injecting model.layers.13.ffn_norm as default
Injecting model.layers.14 as default
Injecting model.layers.14.attention as default
Injecting model.layers.14.attention.wqkv as default
Injecting model.layers.14.attention.wo as default
Injecting model.layers.14.attention.rotary_emb as default
Injecting model.layers.14.feed_forward as default
Injecting model.layers.14.feed_forward.w1 as default
Injecting model.layers.14.feed_forward.w3 as default
Injecting model.layers.14.feed_forward.w2 as default
Injecting model.layers.14.feed_forward.act_fn as default
Injecting model.layers.14.attention_norm as default
Injecting model.layers.14.ffn_norm as default
Injecting model.layers.15 as default
Injecting model.layers.15.attention as default
Injecting model.layers.15.attention.wqkv as default
Injecting model.layers.15.attention.wo as default
Injecting model.layers.15.attention.rotary_emb as default
Injecting model.layers.15.feed_forward as default
Injecting model.layers.15.feed_forward.w1 as default
Injecting model.layers.15.feed_forward.w3 as default
Injecting model.layers.15.feed_forward.w2 as default
Injecting model.layers.15.feed_forward.act_fn as default
Injecting model.layers.15.attention_norm as default
Injecting model.layers.15.ffn_norm as default
Injecting model.layers.16 as default
Injecting model.layers.16.attention as default
Injecting model.layers.16.attention.wqkv as default
Injecting model.layers.16.attention.wo as default
Injecting model.layers.16.attention.rotary_emb as default
Injecting model.layers.16.feed_forward as default
Injecting model.layers.16.feed_forward.w1 as default
Injecting model.layers.16.feed_forward.w3 as default
Injecting model.layers.16.feed_forward.w2 as default
Injecting model.layers.16.feed_forward.act_fn as default
Injecting model.layers.16.attention_norm as default
Injecting model.layers.16.ffn_norm as default
Injecting model.layers.17 as default
Injecting model.layers.17.attention as default
Injecting model.layers.17.attention.wqkv as default
Injecting model.layers.17.attention.wo as default
Injecting model.layers.17.attention.rotary_emb as default
Injecting model.layers.17.feed_forward as default
Injecting model.layers.17.feed_forward.w1 as default
Injecting model.layers.17.feed_forward.w3 as default
Injecting model.layers.17.feed_forward.w2 as default
Injecting model.layers.17.feed_forward.act_fn as default
Injecting model.layers.17.attention_norm as default
Injecting model.layers.17.ffn_norm as default
Injecting model.layers.18 as default
Injecting model.layers.18.attention as default
Injecting model.layers.18.attention.wqkv as default
Injecting model.layers.18.attention.wo as default
Injecting model.layers.18.attention.rotary_emb as default
Injecting model.layers.18.feed_forward as default
Injecting model.layers.18.feed_forward.w1 as default
Injecting model.layers.18.feed_forward.w3 as default
Injecting model.layers.18.feed_forward.w2 as default
Injecting model.layers.18.feed_forward.act_fn as default
Injecting model.layers.18.attention_norm as default
Injecting model.layers.18.ffn_norm as default
Injecting model.layers.19 as default
Injecting model.layers.19.attention as default
Injecting model.layers.19.attention.wqkv as default
Injecting model.layers.19.attention.wo as default
Injecting model.layers.19.attention.rotary_emb as default
Injecting model.layers.19.feed_forward as default
Injecting model.layers.19.feed_forward.w1 as default
Injecting model.layers.19.feed_forward.w3 as default
Injecting model.layers.19.feed_forward.w2 as default
Injecting model.layers.19.feed_forward.act_fn as default
Injecting model.layers.19.attention_norm as default
Injecting model.layers.19.ffn_norm as default
Injecting model.layers.20 as default
Injecting model.layers.20.attention as default
Injecting model.layers.20.attention.wqkv as default
Injecting model.layers.20.attention.wo as default
Injecting model.layers.20.attention.rotary_emb as default
Injecting model.layers.20.feed_forward as default
Injecting model.layers.20.feed_forward.w1 as default
Injecting model.layers.20.feed_forward.w3 as default
Injecting model.layers.20.feed_forward.w2 as default
Injecting model.layers.20.feed_forward.act_fn as default
Injecting model.layers.20.attention_norm as default
Injecting model.layers.20.ffn_norm as default
Injecting model.layers.21 as default
Injecting model.layers.21.attention as default
Injecting model.layers.21.attention.wqkv as default
Injecting model.layers.21.attention.wo as default
Injecting model.layers.21.attention.rotary_emb as default
Injecting model.layers.21.feed_forward as default
Injecting model.layers.21.feed_forward.w1 as default
Injecting model.layers.21.feed_forward.w3 as default
Injecting model.layers.21.feed_forward.w2 as default
Injecting model.layers.21.feed_forward.act_fn as default
Injecting model.layers.21.attention_norm as default
Injecting model.layers.21.ffn_norm as default
Injecting model.layers.22 as default
Injecting model.layers.22.attention as default
Injecting model.layers.22.attention.wqkv as default
Injecting model.layers.22.attention.wo as default
Injecting model.layers.22.attention.rotary_emb as default
Injecting model.layers.22.feed_forward as default
Injecting model.layers.22.feed_forward.w1 as default
Injecting model.layers.22.feed_forward.w3 as default
Injecting model.layers.22.feed_forward.w2 as default
Injecting model.layers.22.feed_forward.act_fn as default
Injecting model.layers.22.attention_norm as default
Injecting model.layers.22.ffn_norm as default
Injecting model.layers.23 as default
Injecting model.layers.23.attention as default
Injecting model.layers.23.attention.wqkv as default
Injecting model.layers.23.attention.wo as default
Injecting model.layers.23.attention.rotary_emb as default
Injecting model.layers.23.feed_forward as default
Injecting model.layers.23.feed_forward.w1 as default
Injecting model.layers.23.feed_forward.w3 as default
Injecting model.layers.23.feed_forward.w2 as default
Injecting model.layers.23.feed_forward.act_fn as default
Injecting model.layers.23.attention_norm as default
Injecting model.layers.23.ffn_norm as default
Injecting model.layers.24 as default
Injecting model.layers.24.attention as default
Injecting model.layers.24.attention.wqkv as default
Injecting model.layers.24.attention.wo as default
Injecting model.layers.24.attention.rotary_emb as default
Injecting model.layers.24.feed_forward as default
Injecting model.layers.24.feed_forward.w1 as default
Injecting model.layers.24.feed_forward.w3 as default
Injecting model.layers.24.feed_forward.w2 as default
Injecting model.layers.24.feed_forward.act_fn as default
Injecting model.layers.24.attention_norm as default
Injecting model.layers.24.ffn_norm as default
Injecting model.layers.25 as default
Injecting model.layers.25.attention as default
Injecting model.layers.25.attention.wqkv as default
Injecting model.layers.25.attention.wo as default
Injecting model.layers.25.attention.rotary_emb as default
Injecting model.layers.25.feed_forward as default
Injecting model.layers.25.feed_forward.w1 as default
Injecting model.layers.25.feed_forward.w3 as default
Injecting model.layers.25.feed_forward.w2 as default
Injecting model.layers.25.feed_forward.act_fn as default
Injecting model.layers.25.attention_norm as default
Injecting model.layers.25.ffn_norm as default
Injecting model.layers.26 as default
Injecting model.layers.26.attention as default
Injecting model.layers.26.attention.wqkv as default
Injecting model.layers.26.attention.wo as default
Injecting model.layers.26.attention.rotary_emb as default
Injecting model.layers.26.feed_forward as default
Injecting model.layers.26.feed_forward.w1 as default
Injecting model.layers.26.feed_forward.w3 as default
Injecting model.layers.26.feed_forward.w2 as default
Injecting model.layers.26.feed_forward.act_fn as default
Injecting model.layers.26.attention_norm as default
Injecting model.layers.26.ffn_norm as default
Injecting model.layers.27 as default
Injecting model.layers.27.attention as default
Injecting model.layers.27.attention.wqkv as default
Injecting model.layers.27.attention.wo as default
Injecting model.layers.27.attention.rotary_emb as default
Injecting model.layers.27.feed_forward as default
Injecting model.layers.27.feed_forward.w1 as default
Injecting model.layers.27.feed_forward.w3 as default
Injecting model.layers.27.feed_forward.w2 as default
Injecting model.layers.27.feed_forward.act_fn as default
Injecting model.layers.27.attention_norm as default
Injecting model.layers.27.ffn_norm as default
Injecting model.layers.28 as default
Injecting model.layers.28.attention as default
Injecting model.layers.28.attention.wqkv as default
Injecting model.layers.28.attention.wo as default
Injecting model.layers.28.attention.rotary_emb as default
Injecting model.layers.28.feed_forward as default
Injecting model.layers.28.feed_forward.w1 as default
Injecting model.layers.28.feed_forward.w3 as default
Injecting model.layers.28.feed_forward.w2 as default
Injecting model.layers.28.feed_forward.act_fn as default
Injecting model.layers.28.attention_norm as default
Injecting model.layers.28.ffn_norm as default
Injecting model.layers.29 as default
Injecting model.layers.29.attention as default
Injecting model.layers.29.attention.wqkv as default
Injecting model.layers.29.attention.wo as default
Injecting model.layers.29.attention.rotary_emb as default
Injecting model.layers.29.feed_forward as default
Injecting model.layers.29.feed_forward.w1 as default
Injecting model.layers.29.feed_forward.w3 as default
Injecting model.layers.29.feed_forward.w2 as default
Injecting model.layers.29.feed_forward.act_fn as default
Injecting model.layers.29.attention_norm as default
Injecting model.layers.29.ffn_norm as default
Injecting model.layers.30 as default
Injecting model.layers.30.attention as default
Injecting model.layers.30.attention.wqkv as default
Injecting model.layers.30.attention.wo as default
Injecting model.layers.30.attention.rotary_emb as default
Injecting model.layers.30.feed_forward as default
Injecting model.layers.30.feed_forward.w1 as default
Injecting model.layers.30.feed_forward.w3 as default
Injecting model.layers.30.feed_forward.w2 as default
Injecting model.layers.30.feed_forward.act_fn as default
Injecting model.layers.30.attention_norm as default
Injecting model.layers.30.ffn_norm as default
Injecting model.layers.31 as default
Injecting model.layers.31.attention as default
Injecting model.layers.31.attention.wqkv as default
Injecting model.layers.31.attention.wo as default
Injecting model.layers.31.attention.rotary_emb as default
Injecting model.layers.31.feed_forward as default
Injecting model.layers.31.feed_forward.w1 as default
Injecting model.layers.31.feed_forward.w3 as default
Injecting model.layers.31.feed_forward.w2 as default
Injecting model.layers.31.feed_forward.act_fn as default
Injecting model.layers.31.attention_norm as default
Injecting model.layers.31.ffn_norm as default
Injecting model.norm as default
Injecting output as default
Traceback (most recent call last):
File "/host/ktransformers-0.2.1/./ktransformers/local_chat.py", line 179, in
fire.Fire(local_chat)
File "/usr/local/lib/python3.10/dist-packages/fire/core.py", line 135, in Fire
component_trace = _Fire(component, args, parsed_flag_args, context, name)
File "/usr/local/lib/python3.10/dist-packages/fire/core.py", line 468, in _Fire
component, remaining_args = _CallAndUpdateTrace(
File "/usr/local/lib/python3.10/dist-packages/fire/core.py", line 684, in _CallAndUpdateTrace
component = fn(*varargs, **kwargs)
File "/host/ktransformers-0.2.1/./ktransformers/local_chat.py", line 110, in local_chat
optimize_and_load_gguf(model, optimize_rule_path, gguf_path, config)
File "/host/ktransformers-0.2.1/./ktransformers/optimize/optimize.py", line 129, in optimize_and_load_gguf
load_weights(module, gguf_loader)
File "/host/ktransformers-0.2.1/./ktransformers/util/utils.py", line 85, in load_weights
load_weights(child, gguf_loader, prefix+name+".")
File "/host/ktransformers-0.2.1/./ktransformers/util/utils.py", line 85, in load_weights
load_weights(child, gguf_loader, prefix+name+".")
File "/host/ktransformers-0.2.1/./ktransformers/util/utils.py", line 83, in load_weights
load_cur_state_dict(module, gguf_loader, prefix)
File "/host/ktransformers-0.2.1/./ktransformers/util/utils.py", line 78, in load_cur_state_dict
raise Exception(f"can't find {translated_key} in GGUF file!")
Exception: can't find model.tok_embeddings.weight in GGUF file!
`
@ubergarm commented on GitHub (Feb 23, 2025):
同学,可以查看这边关于如何运行小规模密集型模型(例如「internlm2_5-7b」)的技术问题。
https://github.com/kvcache-ai/ktransformers/issues/611#issuecomment-2676984590
Hey bud, check out this issue over here for running small dense models like
internlm2_5-7bon other inference engines:@ayttop commented on GitHub (Jun 15, 2025):
mixtral 8*7b run or no with
ktransformers???
@ayttop commented on GitHub (Jun 15, 2025):
Moonlight?????
Qwen2-57B-A14B?????
it suport in ktransformers?????
@ayttop commented on GitHub (Jun 15, 2025):
If anyone has tried or knows,
Moonlight?????
Qwen2-57B-A14B?????
mixtral 8*7b ؟؟؟؟
there is a Yaml for them, but there is a Yaml called Internlm2_5-7b-Chat-1m.yaml, and it doesn't work, as far as I know.
@ayttop commented on GitHub (Jun 15, 2025):
I want to try the library, not the models.
I can't load large models.
@ayttop commented on GitHub (Jun 15, 2025):
https://github.com/kvcache-ai/ktransformers/issues/1390
@Atream commented on GitHub (Jun 16, 2025):
We previously supported local inference of InternLM's 1M context, but the model's support has been damaged with the version update. We'll fix it asap.