Browse Source

add prompt back to parser

Jeffrey Morgan 1 năm trước cách đây
mục cha
commit
d59b164fa2
2 tập tin đã thay đổi với 2 bổ sung2 xóa
  1. 1 1
      parser/parser.go
  2. 1 1
      server/images.go

+ 1 - 1
parser/parser.go

@@ -39,7 +39,7 @@ func Parse(reader io.Reader) ([]Command, error) {
 			command.Args = string(fields[1])
 			// copy command for validation
 			modelCommand = command
-		case "LICENSE", "TEMPLATE", "SYSTEM":
+		case "LICENSE", "TEMPLATE", "SYSTEM", "PROMPT":
 			command.Name = string(bytes.ToLower(fields[0]))
 			command.Args = string(fields[1])
 		case "PARAMETER":

+ 1 - 1
server/images.go

@@ -250,7 +250,7 @@ func CreateModel(name string, path string, fn func(status string)) error {
 					layers = append(layers, newLayer)
 				}
 			}
-		case "license", "template", "system":
+		case "license", "template", "system", "prompt":
 			fn(fmt.Sprintf("creating %s layer", c.Name))
 			// remove the prompt layer if one exists
 			mediaType := fmt.Sprintf("application/vnd.ollama.image.%s", c.Name)