Skip to content

GitLab

Commit ca210ba4 authored by Michael Yang's avatar Michael Yang
Browse files

handle vnd.ollama.image.prompt for compat

parent df146c41
Hide whitespace changes
Inline Side-by-side
Showing with 11 additions and 0 deletions
server/images.go
View file @ ca210ba4
...@@ -39,6 +39,9 @@ func (m *Model) Prompt(request api.GenerateRequest) (string, error) { ...@@ -39,6 +39,9 @@ func (m *Model) Prompt(request api.GenerateRequest) (string, error) {
var vars struct { var vars struct {
System string System string
Prompt string Prompt string
// deprecated: versions <= 0.0.7 used this to omit the system prompt
Context []int
} }
vars.System = m.System vars.System = m.System
...@@ -149,6 +152,14 @@ func GetModel(name string) (*Model, error) { ...@@ -149,6 +152,14 @@ func GetModel(name string) (*Model, error) {
} }
model.System = string(bts) model.System = string(bts)
case "application/vnd.ollama.image.prompt":
log.Printf("PROMPT is deprecated. Please use TEMPLATE and SYSTEM instead.")
bts, err := os.ReadFile(filename)
if err != nil {
return nil, err
}
model.Template = string(bts)
case "application/vnd.ollama.image.params": case "application/vnd.ollama.image.params":
params, err := os.Open(filename) params, err := os.Open(filename)
if err != nil { if err != nil {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment