convert: import support for command-r models from safetensors (#6063)

---------

Co-authored-by: Patrick Devine <patrick@infrahq.com>
This commit is contained in:
Josh
2025-01-15 16:31:22 -08:00
committed by GitHub
parent a041b4df7c
commit 93a8daf285
10 changed files with 503 additions and 0 deletions

View File

@@ -191,6 +191,8 @@ func ConvertModel(fsys fs.FS, ws io.WriteSeeker) error {
conv = &qwen2Model{}
case "BertModel":
conv = &bertModel{}
case "CohereForCausalLM":
conv = &commandrModel{}
default:
return errors.New("unsupported architecture")
}