Skip to content

Commit e39e0b8

Browse files
authored
Fix Mistral3/Gemma3 model hparams init
1 parent 053b3f9 commit e39e0b8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

convert_hf_to_gguf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1752,7 +1752,7 @@ class Mistral3Model(LlamaModel):
17521752

17531753
# we need to merge the text_config into the root level of hparams
17541754
def __init__(self, *args, **kwargs):
1755-
hparams = Model.load_hparams(kwargs["dir_model"])
1755+
hparams = Model.load_hparams(args[0])
17561756
if "text_config" in hparams:
17571757
hparams = {**hparams, **hparams["text_config"]}
17581758
kwargs["hparams"] = hparams
@@ -3385,7 +3385,7 @@ class Gemma3Model(Model):
33853385

33863386
# we need to merge the text_config into the root level of hparams
33873387
def __init__(self, *args, **kwargs):
3388-
hparams = Model.load_hparams(kwargs["dir_model"])
3388+
hparams = Model.load_hparams(args[0])
33893389
if "text_config" in hparams:
33903390
hparams = {**hparams, **hparams["text_config"]}
33913391
kwargs["hparams"] = hparams

0 commit comments

Comments
 (0)