baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

handle vnd.ollama.image.prompt for compat

Browse source
This commit is contained in:
Michael Yang 2023-07-19 19:43:00 -07:00
parent df146c41e2
commit ca210ba480
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
server/images.go
View file

@ -39,6 +39,9 @@ func (m *Model) Prompt(request api.GenerateRequest) (string, error) {
var vars struct {
System string
Prompt string
// deprecated: versions <= 0.0.7 used this to omit the system prompt
Context []int
}
vars.System = m.System
@ -149,6 +152,14 @@ func GetModel(name string) (*Model, error) {
}
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":
params, err := os.Open(filename)
if err != nil {