File tree 1 file changed +2
-2
lines changed
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -1752,7 +1752,7 @@ class Mistral3Model(LlamaModel):
1752
1752
1753
1753
# we need to merge the text_config into the root level of hparams
1754
1754
def __init__ (self , * args , ** kwargs ):
1755
- hparams = Model .load_hparams (kwargs [ "dir_model" ])
1755
+ hparams = Model .load_hparams (args [ 0 ])
1756
1756
if "text_config" in hparams :
1757
1757
hparams = {** hparams , ** hparams ["text_config" ]}
1758
1758
kwargs ["hparams" ] = hparams
@@ -3385,7 +3385,7 @@ class Gemma3Model(Model):
3385
3385
3386
3386
# we need to merge the text_config into the root level of hparams
3387
3387
def __init__ (self , * args , ** kwargs ):
3388
- hparams = Model .load_hparams (kwargs [ "dir_model" ])
3388
+ hparams = Model .load_hparams (args [ 0 ])
3389
3389
if "text_config" in hparams :
3390
3390
hparams = {** hparams , ** hparams ["text_config" ]}
3391
3391
kwargs ["hparams" ] = hparams
You can’t perform that action at this time.
0 commit comments