Skip to content

Commit d1e8590

Browse files
author
Oliver
committed
feat(conf): add flag -hf ab default for extra models
1 parent 43de10a commit d1e8590

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

conf/write.go

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -225,9 +225,12 @@ func (cfg *Cfg) setSwapModels() {
225225
cfg.addModelCfg(model, "${cmd-common} "+flags, commonMC)
226226
case strings.HasPrefix(flags, "--fim-"):
227227
cfg.addModelCfg(model, "${cmd-common} "+flags, fimMC)
228-
default:
228+
case strings.Contains(flags, "-m "), strings.Contains(flags, "-hf "):
229229
cfg.addModelCfg(model, "${cmd-common} "+flags, commonMC)
230230
cfg.addModelCfg("GI_"+model, "${cmd-goinfer} "+flags, goinferMC)
231+
default:
232+
cfg.addModelCfg(model, "${cmd-common} -hf "+model+" "+flags, commonMC)
233+
cfg.addModelCfg("GI_"+model, "${cmd-goinfer} -hf "+model+" "+flags, goinferMC)
231234
}
232235
}
233236

0 commit comments

Comments
 (0)