Skip to content

Commit d6bd4e0

Browse files
authored
check model_type instead
1 parent 926b566 commit d6bd4e0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

convert_hf_to_gguf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -558,7 +558,7 @@ def set_gguf_parameters(self):
558558
if (head_dim := self.hparams.get("head_dim")) is not None:
559559
# Workaround for incorrect AutoConfig value for DeepSeekV3 (is set correctly in DeepSeekV2Model class)
560560
# https://github.com/huggingface/transformers/blob/19224c3642705c5b6988c9f5f4251f83323d05ae/src/transformers/models/deepseek_v3/configuration_deepseek_v3.py#L210
561-
if self.hparams.get("qk_rope_head_dim") != head_dim:
561+
if self.hparams.get("model_type") != "deepseek_v3":
562562
self.gguf_writer.add_key_length(head_dim)
563563
self.gguf_writer.add_value_length(head_dim)
564564

0 commit comments

Comments
 (0)