server: validate template (#5734)

add template validation to modelfile
This commit is contained in:
Josh
2024-07-19 15:24:29 -07:00
committed by GitHub
parent c57317cbf0
commit e8b954c646
3 changed files with 53 additions and 3 deletions

View File

@@ -492,6 +492,12 @@ func CreateModel(ctx context.Context, name model.Name, modelFileDir, quantizatio
layers = append(layers, baseLayer.Layer)
}
case "license", "template", "system":
if c.Name == "template" {
if _, err := template.Parse(c.Args); err != nil {
return fmt.Errorf("%w: %s", errBadTemplate, err)
}
}
if c.Name != "license" {
// replace
layers = slices.DeleteFunc(layers, func(layer *Layer) bool {