I successfully compiled and started ktransformers, but I'm stuck here. Has anyone encountered this?

[zxfingithub]

**python ktransformers/server/main.py --model_path deepseek-ai/DeepSeek-V3  --model_name unsloth/DeepSeek-V3  --gguf_path /home/tztp/ktransformers/models/DeepSeek-V3 --cache_lens 32768 --chunk_size 256 --max_batch_size 4   --backend_type balance_serve --port 10002 --max_new_tokens 8192**
2025-04-23 19:56:33,056 - INFO - flashinfer.jit: Prebuilt kernels not found, using JIT backend
found flashinfer
set start method
tokenizer_config.json: 100%|█████████████████████████████████████████████████████████████████████████| 3.13k/3.13k [00:00<00:00, 30.9MB/s]
tokenizer.json: 100%|████████████████████████████████████████████████████████████████████████████████| 7.85M/7.85M [00:04<00:00, 1.71MB/s]
Connected to server at tcp://localhost:52319
2025-04-23 19:56:43,972 - INFO - flashinfer.jit: Prebuilt kernels not found, using JIT backend
found flashinfer
start method already set to spawn
Connected to server at tcp://localhost:52319
config.json: 100%|███████████████████████████████████████████████████████████████████████████████████| 1.66k/1.66k [00:00<00:00, 18.7MB/s]
configuration_deepseek.py: 100%|█████████████████████████████████████████████████████████████████████| 9.90k/9.90k [00:00<00:00, 2.04MB/s]
A new version of the following files was downloaded from https://huggingface.co/deepseek-ai/DeepSeek-V3:
- configuration_deepseek.py
. Make sure to double-check they do not contain any added malicious code. To avoid downloading new versions of the code file, you can pin a revision.
falsh attn not found
Injecting model as ktransformers.operators.models . KDeepseekV2Model
Injecting model.embed_tokens as default
Injecting model.layers as default
Injecting model.layers.0 as default
Injecting model.layers.0.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.0.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.0.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.0.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.0.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.0.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.0.self_attn.kv_b_proj as default
Injecting model.layers.0.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.0.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.0.mlp as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.0.mlp.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.0.mlp.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.0.mlp.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.0.mlp.act_fn as default
Injecting model.layers.0.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.0.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.1 as default
Injecting model.layers.1.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.1.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.1.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.1.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.1.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.1.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.1.self_attn.kv_b_proj as default
Injecting model.layers.1.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.1.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.1.mlp as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.1.mlp.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.1.mlp.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.1.mlp.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.1.mlp.act_fn as default
Injecting model.layers.1.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.1.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.2 as default
Injecting model.layers.2.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.2.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.2.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.2.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.2.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.2.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.2.self_attn.kv_b_proj as default
Injecting model.layers.2.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.2.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.2.mlp as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.2.mlp.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.2.mlp.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.2.mlp.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.2.mlp.act_fn as default
Injecting model.layers.2.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.2.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.3 as default
Injecting model.layers.3.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.3.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.3.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.3.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.3.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.3.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.3.self_attn.kv_b_proj as default
Injecting model.layers.3.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.3.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.3.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.3.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.3.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.3.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.3.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.3.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.3.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.3.mlp.shared_experts.act_fn as default
Injecting model.layers.3.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.3.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.4 as default
Injecting model.layers.4.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.4.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.4.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.4.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.4.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.4.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.4.self_attn.kv_b_proj as default
Injecting model.layers.4.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.4.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.4.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.4.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.4.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.4.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.4.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.4.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.4.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.4.mlp.shared_experts.act_fn as default
Injecting model.layers.4.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.4.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.5 as default
Injecting model.layers.5.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.5.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.5.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.5.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.5.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.5.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.5.self_attn.kv_b_proj as default
Injecting model.layers.5.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.5.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.5.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.5.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.5.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.5.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.5.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.5.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.5.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.5.mlp.shared_experts.act_fn as default
Injecting model.layers.5.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.5.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.6 as default
Injecting model.layers.6.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.6.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.6.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.6.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.6.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.6.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.6.self_attn.kv_b_proj as default
Injecting model.layers.6.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.6.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.6.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.6.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.6.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.6.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.6.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.6.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.6.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.6.mlp.shared_experts.act_fn as default
Injecting model.layers.6.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.6.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.7 as default
Injecting model.layers.7.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.7.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.7.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.7.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.7.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.7.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.7.self_attn.kv_b_proj as default
Injecting model.layers.7.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.7.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.7.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.7.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.7.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.7.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.7.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.7.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.7.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.7.mlp.shared_experts.act_fn as default
Injecting model.layers.7.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.7.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.8 as default
Injecting model.layers.8.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.8.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.8.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.8.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.8.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.8.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.8.self_attn.kv_b_proj as default
Injecting model.layers.8.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.8.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.8.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.8.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.8.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.8.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.8.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.8.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.8.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.8.mlp.shared_experts.act_fn as default
Injecting model.layers.8.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.8.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.9 as default
Injecting model.layers.9.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.9.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.9.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.9.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.9.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.9.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.9.self_attn.kv_b_proj as default
Injecting model.layers.9.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.9.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.9.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.9.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.9.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.9.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.9.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.9.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.9.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.9.mlp.shared_experts.act_fn as default
Injecting model.layers.9.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.9.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.10 as default
Injecting model.layers.10.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.10.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.10.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.10.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.10.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.10.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.10.self_attn.kv_b_proj as default
Injecting model.layers.10.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.10.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.10.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.10.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.10.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.10.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.10.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.10.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.10.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.10.mlp.shared_experts.act_fn as default
Injecting model.layers.10.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.10.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.11 as default
Injecting model.layers.11.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.11.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.11.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.11.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.11.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.11.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.11.self_attn.kv_b_proj as default
Injecting model.layers.11.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.11.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.11.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.11.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.11.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.11.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.11.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.11.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.11.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.11.mlp.shared_experts.act_fn as default
Injecting model.layers.11.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.11.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.12 as default
Injecting model.layers.12.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.12.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.12.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.12.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.12.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.12.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.12.self_attn.kv_b_proj as default
Injecting model.layers.12.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.12.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.12.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.12.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.12.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.12.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.12.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.12.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.12.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.12.mlp.shared_experts.act_fn as default
Injecting model.layers.12.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.12.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.13 as default
Injecting model.layers.13.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.13.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.13.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.13.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.13.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.13.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.13.self_attn.kv_b_proj as default
Injecting model.layers.13.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.13.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.13.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.13.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.13.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.13.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.13.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.13.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.13.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.13.mlp.shared_experts.act_fn as default
Injecting model.layers.13.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.13.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.14 as default
Injecting model.layers.14.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.14.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.14.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.14.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.14.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.14.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.14.self_attn.kv_b_proj as default
Injecting model.layers.14.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.14.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.14.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.14.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.14.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.14.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.14.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.14.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.14.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.14.mlp.shared_experts.act_fn as default
Injecting model.layers.14.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.14.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.15 as default
Injecting model.layers.15.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.15.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.15.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.15.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.15.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.15.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.15.self_attn.kv_b_proj as default
Injecting model.layers.15.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.15.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.15.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.15.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.15.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.15.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.15.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.15.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.15.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.15.mlp.shared_experts.act_fn as default
Injecting model.layers.15.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.15.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.16 as default
Injecting model.layers.16.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.16.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.16.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.16.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.16.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.16.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.16.self_attn.kv_b_proj as default
Injecting model.layers.16.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.16.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.16.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.16.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.16.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.16.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.16.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.16.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.16.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.16.mlp.shared_experts.act_fn as default
Injecting model.layers.16.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.16.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.17 as default
Injecting model.layers.17.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.17.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.17.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.17.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.17.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.17.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.17.self_attn.kv_b_proj as default
Injecting model.layers.17.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.17.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.17.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.17.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.17.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.17.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.17.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.17.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.17.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.17.mlp.shared_experts.act_fn as default
Injecting model.layers.17.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.17.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.18 as default
Injecting model.layers.18.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.18.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.18.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.18.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.18.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.18.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.18.self_attn.kv_b_proj as default
Injecting model.layers.18.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.18.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.18.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.18.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.18.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.18.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.18.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.18.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.18.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.18.mlp.shared_experts.act_fn as default
Injecting model.layers.18.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.18.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.19 as default
Injecting model.layers.19.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.19.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.19.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.19.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.19.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.19.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.19.self_attn.kv_b_proj as default
Injecting model.layers.19.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.19.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.19.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.19.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.19.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.19.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.19.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.19.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.19.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.19.mlp.shared_experts.act_fn as default
Injecting model.layers.19.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.19.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.20 as default
Injecting model.layers.20.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.20.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.20.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.20.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.20.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.20.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.20.self_attn.kv_b_proj as default
Injecting model.layers.20.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.20.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.20.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.20.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.20.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.20.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.20.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.20.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.20.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.20.mlp.shared_experts.act_fn as default
Injecting model.layers.20.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.20.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.21 as default
Injecting model.layers.21.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.21.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.21.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.21.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.21.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.21.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.21.self_attn.kv_b_proj as default
Injecting model.layers.21.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.21.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.21.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.21.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.21.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.21.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.21.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.21.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.21.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.21.mlp.shared_experts.act_fn as default
Injecting model.layers.21.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.21.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.22 as default
Injecting model.layers.22.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.22.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.22.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.22.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.22.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.22.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.22.self_attn.kv_b_proj as default
Injecting model.layers.22.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.22.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.22.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.22.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.22.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.22.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.22.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.22.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.22.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.22.mlp.shared_experts.act_fn as default
Injecting model.layers.22.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.22.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.23 as default
Injecting model.layers.23.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.23.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.23.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.23.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.23.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.23.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.23.self_attn.kv_b_proj as default
Injecting model.layers.23.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.23.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.23.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.23.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.23.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.23.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.23.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.23.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.23.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.23.mlp.shared_experts.act_fn as default
Injecting model.layers.23.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.23.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.24 as default
Injecting model.layers.24.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.24.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.24.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.24.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.24.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.24.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.24.self_attn.kv_b_proj as default
Injecting model.layers.24.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.24.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.24.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.24.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.24.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.24.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.24.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.24.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.24.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.24.mlp.shared_experts.act_fn as default
Injecting model.layers.24.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.24.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.25 as default
Injecting model.layers.25.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.25.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.25.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.25.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.25.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.25.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.25.self_attn.kv_b_proj as default
Injecting model.layers.25.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.25.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.25.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.25.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.25.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.25.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.25.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.25.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.25.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.25.mlp.shared_experts.act_fn as default
Injecting model.layers.25.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.25.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.26 as default
Injecting model.layers.26.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.26.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.26.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.26.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.26.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.26.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.26.self_attn.kv_b_proj as default
Injecting model.layers.26.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.26.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.26.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.26.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.26.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.26.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.26.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.26.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.26.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.26.mlp.shared_experts.act_fn as default
Injecting model.layers.26.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.26.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.27 as default
Injecting model.layers.27.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.27.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.27.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.27.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.27.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.27.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.27.self_attn.kv_b_proj as default
Injecting model.layers.27.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.27.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.27.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.27.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.27.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.27.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.27.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.27.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.27.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.27.mlp.shared_experts.act_fn as default
Injecting model.layers.27.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.27.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.28 as default
Injecting model.layers.28.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.28.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.28.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.28.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.28.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.28.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.28.self_attn.kv_b_proj as default
Injecting model.layers.28.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.28.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.28.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.28.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.28.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.28.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.28.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.28.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.28.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.28.mlp.shared_experts.act_fn as default
Injecting model.layers.28.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.28.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.29 as default
Injecting model.layers.29.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.29.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.29.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.29.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.29.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.29.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.29.self_attn.kv_b_proj as default
Injecting model.layers.29.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.29.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.29.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.29.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.29.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.29.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.29.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.29.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.29.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.29.mlp.shared_experts.act_fn as default
Injecting model.layers.29.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.29.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.30 as default
Injecting model.layers.30.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.30.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.30.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.30.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.30.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.30.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.30.self_attn.kv_b_proj as default
Injecting model.layers.30.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.30.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.30.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.30.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.30.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.30.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.30.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.30.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.30.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.30.mlp.shared_experts.act_fn as default
Injecting model.layers.30.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.30.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.31 as default
Injecting model.layers.31.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.31.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.31.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.31.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.31.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.31.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.31.self_attn.kv_b_proj as default
Injecting model.layers.31.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.31.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.31.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.31.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.31.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.31.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.31.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.31.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.31.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.31.mlp.shared_experts.act_fn as default
Injecting model.layers.31.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.31.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.32 as default
Injecting model.layers.32.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.32.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.32.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.32.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.32.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.32.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.32.self_attn.kv_b_proj as default
Injecting model.layers.32.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.32.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.32.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.32.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.32.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.32.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.32.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.32.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.32.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.32.mlp.shared_experts.act_fn as default
Injecting model.layers.32.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.32.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.33 as default
Injecting model.layers.33.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.33.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.33.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.33.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.33.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.33.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.33.self_attn.kv_b_proj as default
Injecting model.layers.33.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.33.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.33.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.33.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.33.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.33.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.33.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.33.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.33.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.33.mlp.shared_experts.act_fn as default
Injecting model.layers.33.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.33.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.34 as default
Injecting model.layers.34.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.34.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.34.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.34.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.34.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.34.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.34.self_attn.kv_b_proj as default
Injecting model.layers.34.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.34.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.34.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.34.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.34.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.34.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.34.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.34.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.34.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.34.mlp.shared_experts.act_fn as default
Injecting model.layers.34.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.34.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.35 as default
Injecting model.layers.35.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.35.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.35.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.35.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.35.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.35.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.35.self_attn.kv_b_proj as default
Injecting model.layers.35.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.35.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.35.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.35.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.35.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.35.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.35.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.35.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.35.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.35.mlp.shared_experts.act_fn as default
Injecting model.layers.35.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.35.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.36 as default
Injecting model.layers.36.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.36.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.36.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.36.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.36.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.36.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.36.self_attn.kv_b_proj as default
Injecting model.layers.36.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.36.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.36.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.36.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.36.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.36.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.36.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.36.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.36.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.36.mlp.shared_experts.act_fn as default
Injecting model.layers.36.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.36.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.37 as default
Injecting model.layers.37.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.37.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.37.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.37.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.37.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.37.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.37.self_attn.kv_b_proj as default
Injecting model.layers.37.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.37.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.37.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.37.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.37.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.37.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.37.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.37.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.37.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.37.mlp.shared_experts.act_fn as default
Injecting model.layers.37.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.37.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.38 as default
Injecting model.layers.38.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.38.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.38.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.38.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.38.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.38.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.38.self_attn.kv_b_proj as default
Injecting model.layers.38.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.38.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.38.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.38.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.38.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.38.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.38.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.38.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.38.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.38.mlp.shared_experts.act_fn as default
Injecting model.layers.38.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.38.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.39 as default
Injecting model.layers.39.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.39.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.39.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.39.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.39.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.39.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.39.self_attn.kv_b_proj as default
Injecting model.layers.39.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.39.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.39.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.39.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.39.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.39.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.39.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.39.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.39.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.39.mlp.shared_experts.act_fn as default
Injecting model.layers.39.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.39.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.40 as default
Injecting model.layers.40.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.40.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.40.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.40.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.40.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.40.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.40.self_attn.kv_b_proj as default
Injecting model.layers.40.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.40.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.40.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.40.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.40.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.40.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.40.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.40.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.40.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.40.mlp.shared_experts.act_fn as default
Injecting model.layers.40.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.40.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.41 as default
Injecting model.layers.41.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.41.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.41.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.41.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.41.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.41.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.41.self_attn.kv_b_proj as default
Injecting model.layers.41.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.41.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.41.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.41.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.41.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.41.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.41.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.41.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.41.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.41.mlp.shared_experts.act_fn as default
Injecting model.layers.41.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.41.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.42 as default
Injecting model.layers.42.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.42.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.42.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.42.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.42.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.42.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.42.self_attn.kv_b_proj as default
Injecting model.layers.42.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.42.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.42.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.42.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.42.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.42.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.42.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.42.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.42.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.42.mlp.shared_experts.act_fn as default
Injecting model.layers.42.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.42.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.43 as default
Injecting model.layers.43.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.43.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.43.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.43.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.43.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.43.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.43.self_attn.kv_b_proj as default
Injecting model.layers.43.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.43.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.43.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.43.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.43.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.43.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.43.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.43.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.43.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.43.mlp.shared_experts.act_fn as default
Injecting model.layers.43.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.43.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.44 as default
Injecting model.layers.44.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.44.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.44.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.44.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.44.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.44.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.44.self_attn.kv_b_proj as default
Injecting model.layers.44.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.44.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.44.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.44.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.44.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.44.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.44.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.44.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.44.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.44.mlp.shared_experts.act_fn as default
Injecting model.layers.44.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.44.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.45 as default
Injecting model.layers.45.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.45.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.45.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.45.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.45.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.45.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.45.self_attn.kv_b_proj as default
Injecting model.layers.45.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.45.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.45.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.45.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.45.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.45.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.45.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.45.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.45.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.45.mlp.shared_experts.act_fn as default
Injecting model.layers.45.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.45.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.46 as default
Injecting model.layers.46.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.46.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.46.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.46.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.46.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.46.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.46.self_attn.kv_b_proj as default
Injecting model.layers.46.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.46.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.46.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.46.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.46.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.46.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.46.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.46.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.46.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.46.mlp.shared_experts.act_fn as default
Injecting model.layers.46.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.46.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.47 as default
Injecting model.layers.47.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.47.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.47.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.47.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.47.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.47.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.47.self_attn.kv_b_proj as default
Injecting model.layers.47.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.47.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.47.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.47.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.47.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.47.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.47.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.47.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.47.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.47.mlp.shared_experts.act_fn as default
Injecting model.layers.47.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.47.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.48 as default
Injecting model.layers.48.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.48.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.48.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.48.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.48.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.48.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.48.self_attn.kv_b_proj as default
Injecting model.layers.48.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.48.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.48.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.48.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.48.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.48.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.48.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.48.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.48.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.48.mlp.shared_experts.act_fn as default
Injecting model.layers.48.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.48.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.49 as default
Injecting model.layers.49.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.49.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.49.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.49.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.49.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.49.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.49.self_attn.kv_b_proj as default
Injecting model.layers.49.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.49.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.49.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.49.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.49.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.49.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.49.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.49.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.49.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.49.mlp.shared_experts.act_fn as default
Injecting model.layers.49.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.49.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.50 as default
Injecting model.layers.50.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.50.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.50.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.50.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.50.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.50.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.50.self_attn.kv_b_proj as default
Injecting model.layers.50.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.50.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.50.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.50.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.50.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.50.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.50.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.50.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.50.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.50.mlp.shared_experts.act_fn as default
Injecting model.layers.50.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.50.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.51 as default
Injecting model.layers.51.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.51.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.51.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.51.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.51.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.51.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.51.self_attn.kv_b_proj as default
Injecting model.layers.51.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.51.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.51.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.51.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.51.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.51.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.51.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.51.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.51.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.51.mlp.shared_experts.act_fn as default
Injecting model.layers.51.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.51.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.52 as default
Injecting model.layers.52.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.52.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.52.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.52.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.52.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.52.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.52.self_attn.kv_b_proj as default
Injecting model.layers.52.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.52.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.52.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.52.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.52.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.52.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.52.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.52.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.52.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.52.mlp.shared_experts.act_fn as default
Injecting model.layers.52.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.52.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.53 as default
Injecting model.layers.53.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.53.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.53.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.53.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.53.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.53.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.53.self_attn.kv_b_proj as default
Injecting model.layers.53.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.53.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.53.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.53.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.53.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.53.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.53.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.53.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.53.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.53.mlp.shared_experts.act_fn as default
Injecting model.layers.53.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.53.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.54 as default
Injecting model.layers.54.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.54.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.54.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.54.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.54.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.54.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.54.self_attn.kv_b_proj as default
Injecting model.layers.54.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.54.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.54.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.54.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.54.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.54.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.54.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.54.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.54.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.54.mlp.shared_experts.act_fn as default
Injecting model.layers.54.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.54.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.55 as default
Injecting model.layers.55.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.55.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.55.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.55.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.55.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.55.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.55.self_attn.kv_b_proj as default
Injecting model.layers.55.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.55.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.55.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.55.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.55.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.55.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.55.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.55.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.55.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.55.mlp.shared_experts.act_fn as default
Injecting model.layers.55.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.55.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.56 as default
Injecting model.layers.56.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.56.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.56.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.56.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.56.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.56.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.56.self_attn.kv_b_proj as default
Injecting model.layers.56.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.56.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.56.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.56.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.56.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.56.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.56.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.56.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.56.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.56.mlp.shared_experts.act_fn as default
Injecting model.layers.56.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.56.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.57 as default
Injecting model.layers.57.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.57.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.57.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.57.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.57.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.57.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.57.self_attn.kv_b_proj as default
Injecting model.layers.57.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.57.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.57.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.57.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.57.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.57.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.57.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.57.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.57.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.57.mlp.shared_experts.act_fn as default
Injecting model.layers.57.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.57.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.58 as default
Injecting model.layers.58.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.58.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.58.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.58.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.58.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.58.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.58.self_attn.kv_b_proj as default
Injecting model.layers.58.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.58.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.58.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.58.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.58.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.58.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.58.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.58.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.58.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.58.mlp.shared_experts.act_fn as default
Injecting model.layers.58.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.58.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.59 as default
Injecting model.layers.59.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.59.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.59.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.59.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.59.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.59.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.59.self_attn.kv_b_proj as default
Injecting model.layers.59.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.59.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.59.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.59.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.59.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.59.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.59.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.59.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.59.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.59.mlp.shared_experts.act_fn as default
Injecting model.layers.59.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.59.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.60 as default
Injecting model.layers.60.self_attn as ktransformers.operators.attention . flashinfer_attn
Injecting model.layers.60.self_attn.q_a_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.60.self_attn.q_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.60.self_attn.q_b_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.60.self_attn.kv_a_proj_with_mqa as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.60.self_attn.kv_a_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.60.self_attn.kv_b_proj as default
Injecting model.layers.60.self_attn.o_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.60.self_attn.rotary_emb as ktransformers.operators.RoPE . YarnRotaryEmbeddingV3
Injecting model.layers.60.mlp as ktransformers.operators.experts . KDeepseekV3MoEV2
Injecting model.layers.60.mlp.experts as ktransformers.operators.experts . KTransformersExpertsV2
Injecting model.layers.60.mlp.gate as ktransformers.operators.gate . KMoEGate
Injecting model.layers.60.mlp.shared_experts as ktransformers.operators.mlp . kDeepseekV3MLP
Injecting model.layers.60.mlp.shared_experts.gate_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.60.mlp.shared_experts.up_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.60.mlp.shared_experts.down_proj as ktransformers.operators.linear . KTransformersLinear
Injecting model.layers.60.mlp.shared_experts.act_fn as default
Injecting model.layers.60.input_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.layers.60.post_attention_layernorm as ktransformers.operators.layernorm . RMSNorm
Injecting model.norm as ktransformers.operators.layernorm . RMSNorm
Injecting cache as default
Injecting lm_head as ktransformers.operators.linear . KTransformersLinear
loading token_embd.weight to cpu
loading token_embd.weight with CPU
loading blk.0.attn_q_a_norm.weight to cuda
loading blk.0.attn_kv_a_norm.weight to cuda
loading blk.0.attn_kv_b.weight to cuda:0
loading blk.0.attn_norm.weight to cuda
loading blk.0.ffn_norm.weight to cuda
loading blk.1.attn_q_a_norm.weight to cuda
loading blk.1.attn_kv_a_norm.weight to cuda
loading blk.1.attn_kv_b.weight to cuda:0
loading blk.1.attn_norm.weight to cuda
loading blk.1.ffn_norm.weight to cuda
loading blk.2.attn_q_a_norm.weight to cuda
loading blk.2.attn_kv_a_norm.weight to cuda
loading blk.2.attn_kv_b.weight to cuda:0
loading blk.2.attn_norm.weight to cuda
loading blk.2.ffn_norm.weight to cuda
loading blk.3.attn_q_a_norm.weight to cuda
loading blk.3.attn_kv_a_norm.weight to cuda
loading blk.3.attn_kv_b.weight to cuda:0
loading blk.3.attn_norm.weight to cuda
loading blk.3.ffn_norm.weight to cuda
loading blk.4.attn_q_a_norm.weight to cuda
loading blk.4.attn_kv_a_norm.weight to cuda
loading blk.4.attn_kv_b.weight to cuda:0
loading blk.4.attn_norm.weight to cuda
loading blk.4.ffn_norm.weight to cuda
loading blk.5.attn_q_a_norm.weight to cuda
loading blk.5.attn_kv_a_norm.weight to cuda
loading blk.5.attn_kv_b.weight to cuda:0
loading blk.5.attn_norm.weight to cuda
loading blk.5.ffn_norm.weight to cuda
loading blk.6.attn_q_a_norm.weight to cuda
loading blk.6.attn_kv_a_norm.weight to cuda
loading blk.6.attn_kv_b.weight to cuda:0
loading blk.6.attn_norm.weight to cuda
loading blk.6.ffn_norm.weight to cuda
loading blk.7.attn_q_a_norm.weight to cuda
loading blk.7.attn_kv_a_norm.weight to cuda
loading blk.7.attn_kv_b.weight to cuda:0
loading blk.7.attn_norm.weight to cuda
loading blk.7.ffn_norm.weight to cuda
loading blk.8.attn_q_a_norm.weight to cuda
loading blk.8.attn_kv_a_norm.weight to cuda
loading blk.8.attn_kv_b.weight to cuda:0
loading blk.8.attn_norm.weight to cuda
loading blk.8.ffn_norm.weight to cuda
loading blk.9.attn_q_a_norm.weight to cuda
loading blk.9.attn_kv_a_norm.weight to cuda
loading blk.9.attn_kv_b.weight to cuda:0
loading blk.9.attn_norm.weight to cuda
loading blk.9.ffn_norm.weight to cuda
loading blk.10.attn_q_a_norm.weight to cuda
loading blk.10.attn_kv_a_norm.weight to cuda
loading blk.10.attn_kv_b.weight to cuda:0
loading blk.10.attn_norm.weight to cuda
loading blk.10.ffn_norm.weight to cuda
loading blk.11.attn_q_a_norm.weight to cuda
loading blk.11.attn_kv_a_norm.weight to cuda
loading blk.11.attn_kv_b.weight to cuda:0
loading blk.11.attn_norm.weight to cuda
loading blk.11.ffn_norm.weight to cuda
loading blk.12.attn_q_a_norm.weight to cuda
loading blk.12.attn_kv_a_norm.weight to cuda
loading blk.12.attn_kv_b.weight to cuda:0
loading blk.12.attn_norm.weight to cuda
loading blk.12.ffn_norm.weight to cuda
loading blk.13.attn_q_a_norm.weight to cuda
loading blk.13.attn_kv_a_norm.weight to cuda
loading blk.13.attn_kv_b.weight to cuda:0
loading blk.13.attn_norm.weight to cuda
loading blk.13.ffn_norm.weight to cuda
loading blk.14.attn_q_a_norm.weight to cuda
loading blk.14.attn_kv_a_norm.weight to cuda
loading blk.14.attn_kv_b.weight to cuda:0
loading blk.14.attn_norm.weight to cuda
loading blk.14.ffn_norm.weight to cuda
loading blk.15.attn_q_a_norm.weight to cuda
loading blk.15.attn_kv_a_norm.weight to cuda
loading blk.15.attn_kv_b.weight to cuda:0
loading blk.15.attn_norm.weight to cuda
loading blk.15.ffn_norm.weight to cuda
loading blk.16.attn_q_a_norm.weight to cuda
loading blk.16.attn_kv_a_norm.weight to cuda
loading blk.16.attn_kv_b.weight to cuda:0
loading blk.16.attn_norm.weight to cuda
loading blk.16.ffn_norm.weight to cuda
loading blk.17.attn_q_a_norm.weight to cuda
loading blk.17.attn_kv_a_norm.weight to cuda
loading blk.17.attn_kv_b.weight to cuda:0
loading blk.17.attn_norm.weight to cuda
loading blk.17.ffn_norm.weight to cuda
loading blk.18.attn_q_a_norm.weight to cuda
loading blk.18.attn_kv_a_norm.weight to cuda
loading blk.18.attn_kv_b.weight to cuda:0
loading blk.18.attn_norm.weight to cuda
loading blk.18.ffn_norm.weight to cuda
loading blk.19.attn_q_a_norm.weight to cuda
loading blk.19.attn_kv_a_norm.weight to cuda
loading blk.19.attn_kv_b.weight to cuda:0
loading blk.19.attn_norm.weight to cuda
loading blk.19.ffn_norm.weight to cuda
loading blk.20.attn_q_a_norm.weight to cuda
loading blk.20.attn_kv_a_norm.weight to cuda
loading blk.20.attn_kv_b.weight to cuda:0
loading blk.20.attn_norm.weight to cuda
loading blk.20.ffn_norm.weight to cuda
loading blk.21.attn_q_a_norm.weight to cuda
loading blk.21.attn_kv_a_norm.weight to cuda
loading blk.21.attn_kv_b.weight to cuda:0
loading blk.21.attn_norm.weight to cuda
loading blk.21.ffn_norm.weight to cuda
loading blk.22.attn_q_a_norm.weight to cuda
loading blk.22.attn_kv_a_norm.weight to cuda
loading blk.22.attn_kv_b.weight to cuda:0
loading blk.22.attn_norm.weight to cuda
loading blk.22.ffn_norm.weight to cuda
loading blk.23.attn_q_a_norm.weight to cuda
loading blk.23.attn_kv_a_norm.weight to cuda
loading blk.23.attn_kv_b.weight to cuda:0
loading blk.23.attn_norm.weight to cuda
loading blk.23.ffn_norm.weight to cuda
loading blk.24.attn_q_a_norm.weight to cuda
loading blk.24.attn_kv_a_norm.weight to cuda
loading blk.24.attn_kv_b.weight to cuda:0
loading blk.24.attn_norm.weight to cuda
loading blk.24.ffn_norm.weight to cuda
loading blk.25.attn_q_a_norm.weight to cuda
loading blk.25.attn_kv_a_norm.weight to cuda
loading blk.25.attn_kv_b.weight to cuda:0
loading blk.25.attn_norm.weight to cuda
loading blk.25.ffn_norm.weight to cuda
loading blk.26.attn_q_a_norm.weight to cuda
loading blk.26.attn_kv_a_norm.weight to cuda
loading blk.26.attn_kv_b.weight to cuda:0
loading blk.26.attn_norm.weight to cuda
loading blk.26.ffn_norm.weight to cuda
loading blk.27.attn_q_a_norm.weight to cuda
loading blk.27.attn_kv_a_norm.weight to cuda
loading blk.27.attn_kv_b.weight to cuda:0
loading blk.27.attn_norm.weight to cuda
loading blk.27.ffn_norm.weight to cuda
loading blk.28.attn_q_a_norm.weight to cuda
loading blk.28.attn_kv_a_norm.weight to cuda
loading blk.28.attn_kv_b.weight to cuda:0
loading blk.28.attn_norm.weight to cuda
loading blk.28.ffn_norm.weight to cuda
loading blk.29.attn_q_a_norm.weight to cuda
loading blk.29.attn_kv_a_norm.weight to cuda
loading blk.29.attn_kv_b.weight to cuda:0
loading blk.29.attn_norm.weight to cuda
loading blk.29.ffn_norm.weight to cuda
loading blk.30.attn_q_a_norm.weight to cuda
loading blk.30.attn_kv_a_norm.weight to cuda
loading blk.30.attn_kv_b.weight to cuda:0
loading blk.30.attn_norm.weight to cuda
loading blk.30.ffn_norm.weight to cuda
loading blk.31.attn_q_a_norm.weight to cuda
loading blk.31.attn_kv_a_norm.weight to cuda
loading blk.31.attn_kv_b.weight to cuda:0
loading blk.31.attn_norm.weight to cuda
loading blk.31.ffn_norm.weight to cuda
loading blk.32.attn_q_a_norm.weight to cuda
loading blk.32.attn_kv_a_norm.weight to cuda
loading blk.32.attn_kv_b.weight to cuda:0
loading blk.32.attn_norm.weight to cuda
loading blk.32.ffn_norm.weight to cuda
loading blk.33.attn_q_a_norm.weight to cuda
loading blk.33.attn_kv_a_norm.weight to cuda
loading blk.33.attn_kv_b.weight to cuda:0
loading blk.33.attn_norm.weight to cuda
loading blk.33.ffn_norm.weight to cuda
loading blk.34.attn_q_a_norm.weight to cuda
loading blk.34.attn_kv_a_norm.weight to cuda
loading blk.34.attn_kv_b.weight to cuda:0
loading blk.34.attn_norm.weight to cuda
loading blk.34.ffn_norm.weight to cuda
loading blk.35.attn_q_a_norm.weight to cuda
loading blk.35.attn_kv_a_norm.weight to cuda
loading blk.35.attn_kv_b.weight to cuda:0
loading blk.35.attn_norm.weight to cuda
loading blk.35.ffn_norm.weight to cuda
loading blk.36.attn_q_a_norm.weight to cuda
loading blk.36.attn_kv_a_norm.weight to cuda
loading blk.36.attn_kv_b.weight to cuda:0
loading blk.36.attn_norm.weight to cuda
loading blk.36.ffn_norm.weight to cuda
loading blk.37.attn_q_a_norm.weight to cuda
loading blk.37.attn_kv_a_norm.weight to cuda
loading blk.37.attn_kv_b.weight to cuda:0
loading blk.37.attn_norm.weight to cuda
loading blk.37.ffn_norm.weight to cuda
loading blk.38.attn_q_a_norm.weight to cuda
loading blk.38.attn_kv_a_norm.weight to cuda
loading blk.38.attn_kv_b.weight to cuda:0
loading blk.38.attn_norm.weight to cuda
loading blk.38.ffn_norm.weight to cuda
loading blk.39.attn_q_a_norm.weight to cuda
loading blk.39.attn_kv_a_norm.weight to cuda
loading blk.39.attn_kv_b.weight to cuda:0
loading blk.39.attn_norm.weight to cuda
loading blk.39.ffn_norm.weight to cuda
loading blk.40.attn_q_a_norm.weight to cuda
loading blk.40.attn_kv_a_norm.weight to cuda
loading blk.40.attn_kv_b.weight to cuda:0
loading blk.40.attn_norm.weight to cuda
loading blk.40.ffn_norm.weight to cuda
loading blk.41.attn_q_a_norm.weight to cuda
loading blk.41.attn_kv_a_norm.weight to cuda
loading blk.41.attn_kv_b.weight to cuda:0
loading blk.41.attn_norm.weight to cuda
loading blk.41.ffn_norm.weight to cuda
loading blk.42.attn_q_a_norm.weight to cuda
loading blk.42.attn_kv_a_norm.weight to cuda
loading blk.42.attn_kv_b.weight to cuda:0
loading blk.42.attn_norm.weight to cuda
loading blk.42.ffn_norm.weight to cuda
loading blk.43.attn_q_a_norm.weight to cuda
loading blk.43.attn_kv_a_norm.weight to cuda
loading blk.43.attn_kv_b.weight to cuda:0
loading blk.43.attn_norm.weight to cuda
loading blk.43.ffn_norm.weight to cuda
loading blk.44.attn_q_a_norm.weight to cuda
loading blk.44.attn_kv_a_norm.weight to cuda
loading blk.44.attn_kv_b.weight to cuda:0
loading blk.44.attn_norm.weight to cuda
loading blk.44.ffn_norm.weight to cuda
loading blk.45.attn_q_a_norm.weight to cuda
loading blk.45.attn_kv_a_norm.weight to cuda
loading blk.45.attn_kv_b.weight to cuda:0
loading blk.45.attn_norm.weight to cuda
loading blk.45.ffn_norm.weight to cuda
loading blk.46.attn_q_a_norm.weight to cuda
loading blk.46.attn_kv_a_norm.weight to cuda
loading blk.46.attn_kv_b.weight to cuda:0
loading blk.46.attn_norm.weight to cuda
loading blk.46.ffn_norm.weight to cuda
loading blk.47.attn_q_a_norm.weight to cuda
loading blk.47.attn_kv_a_norm.weight to cuda
loading blk.47.attn_kv_b.weight to cuda:0
loading blk.47.attn_norm.weight to cuda
loading blk.47.ffn_norm.weight to cuda
loading blk.48.attn_q_a_norm.weight to cuda
loading blk.48.attn_kv_a_norm.weight to cuda
loading blk.48.attn_kv_b.weight to cuda:0
loading blk.48.attn_norm.weight to cuda
loading blk.48.ffn_norm.weight to cuda
loading blk.49.attn_q_a_norm.weight to cuda
loading blk.49.attn_kv_a_norm.weight to cuda
loading blk.49.attn_kv_b.weight to cuda:0
loading blk.49.attn_norm.weight to cuda
loading blk.49.ffn_norm.weight to cuda
loading blk.50.attn_q_a_norm.weight to cuda
loading blk.50.attn_kv_a_norm.weight to cuda
loading blk.50.attn_kv_b.weight to cuda:0
loading blk.50.attn_norm.weight to cuda
loading blk.50.ffn_norm.weight to cuda
loading blk.51.attn_q_a_norm.weight to cuda
loading blk.51.attn_kv_a_norm.weight to cuda
loading blk.51.attn_kv_b.weight to cuda:0
loading blk.51.attn_norm.weight to cuda
loading blk.51.ffn_norm.weight to cuda
loading blk.52.attn_q_a_norm.weight to cuda
loading blk.52.attn_kv_a_norm.weight to cuda
loading blk.52.attn_kv_b.weight to cuda:0
loading blk.52.attn_norm.weight to cuda
loading blk.52.ffn_norm.weight to cuda
loading blk.53.attn_q_a_norm.weight to cuda
loading blk.53.attn_kv_a_norm.weight to cuda
loading blk.53.attn_kv_b.weight to cuda:0
loading blk.53.attn_norm.weight to cuda
loading blk.53.ffn_norm.weight to cuda
loading blk.54.attn_q_a_norm.weight to cuda
loading blk.54.attn_kv_a_norm.weight to cuda
loading blk.54.attn_kv_b.weight to cuda:0
loading blk.54.attn_norm.weight to cuda
loading blk.54.ffn_norm.weight to cuda
loading blk.55.attn_q_a_norm.weight to cuda
loading blk.55.attn_kv_a_norm.weight to cuda
loading blk.55.attn_kv_b.weight to cuda:0
loading blk.55.attn_norm.weight to cuda
loading blk.55.ffn_norm.weight to cuda
loading blk.56.attn_q_a_norm.weight to cuda
loading blk.56.attn_kv_a_norm.weight to cuda
loading blk.56.attn_kv_b.weight to cuda:0
loading blk.56.attn_norm.weight to cuda
loading blk.56.ffn_norm.weight to cuda
loading blk.57.attn_q_a_norm.weight to cuda
loading blk.57.attn_kv_a_norm.weight to cuda
loading blk.57.attn_kv_b.weight to cuda:0
loading blk.57.attn_norm.weight to cuda
loading blk.57.ffn_norm.weight to cuda
loading blk.58.attn_q_a_norm.weight to cuda
loading blk.58.attn_kv_a_norm.weight to cuda
loading blk.58.attn_kv_b.weight to cuda:0
loading blk.58.attn_norm.weight to cuda
loading blk.58.ffn_norm.weight to cuda
loading blk.59.attn_q_a_norm.weight to cuda
loading blk.59.attn_kv_a_norm.weight to cuda
loading blk.59.attn_kv_b.weight to cuda:0
loading blk.59.attn_norm.weight to cuda
loading blk.59.ffn_norm.weight to cuda
loading blk.60.attn_q_a_norm.weight to cuda
loading blk.60.attn_kv_a_norm.weight to cuda
loading blk.60.attn_kv_b.weight to cuda:0
loading blk.60.attn_norm.weight to cuda
loading blk.60.ffn_norm.weight to cuda
loading output_norm.weight to cuda
Getting inference context from sched_client.
**![image](https://github.com/user-attachments/assets/048061c7-12ff-465b-95ac-778e7bbcc10c)**

[sz-xuejin]

I got this issue too. How did you solved?

[zxfingithub]

You can refer to #1056 (comment)