Exclude embeddings and output tensor
This commit is contained in:
parent
35ad0fc4ad
commit
5ef493ea1a
|
|
@ -697,8 +697,10 @@ static std::unordered_map<std::string, ggml_type> target_bpw_type(
|
||||||
q &= name.find("time_mix_decay_w2.weight") == std::string::npos;
|
q &= name.find("time_mix_decay_w2.weight") == std::string::npos;
|
||||||
q &= name.find("time_mix_lerp_fused.weight") == std::string::npos;
|
q &= name.find("time_mix_lerp_fused.weight") == std::string::npos;
|
||||||
q &= name.find("attn_rel_b.weight") == std::string::npos;
|
q &= name.find("attn_rel_b.weight") == std::string::npos;
|
||||||
q &= params->quantize_output_tensor || name != "output.weight";
|
|
||||||
q &= !params->only_copy;
|
q &= !params->only_copy;
|
||||||
|
// TODO: Exclude embeddings and output tensors?
|
||||||
|
q &= params->quantize_output_tensor || name != "output.weight";
|
||||||
|
q &= name != name_tn(LLM_TENSOR_TOKEN_EMBD, "weight");
|
||||||
|
|
||||||
return q;
|
return q;
|
||||||
};
|
};
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue