jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

mkdirp new manifest directories

Browse Source
This commit is contained in:
Jeffrey Morgan 2023-07-15 21:39:18 -07:00 committed by Patrick Devine
parent 0573eae4b4
commit 6228a5f39f
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
server/images.go
View File

@ -518,8 +518,7 @@ func PullModel(name, username, password string, fn func(status, digest string, T
manifest, err := pullModelManifest(DefaultRegistry, repoName, tag, username, password) manifest, err := pullModelManifest(DefaultRegistry, repoName, tag, username, password)
if err != nil { if err != nil {
fmt.Errorf("Error: %q", err) return fmt.Errorf("pull model manifest: %q", err)
return err
} }
log.Printf("manifest = %#v", manifest) log.Printf("manifest = %#v", manifest)
@ -556,6 +555,12 @@ func PullModel(name, username, password string, fn func(status, digest string, T
} }
fp := path.Join(home, ".ollama/models/manifests", name) fp := path.Join(home, ".ollama/models/manifests", name)
err = os.MkdirAll(path.Dir(fp), 0o700)
if err != nil {
return fmt.Errorf("make manifests directory: %w", err)
}
err = os.WriteFile(fp, manifestJSON, 0644) err = os.WriteFile(fp, manifestJSON, 0644)
if err != nil { if err != nil {
log.Printf("couldn't write to %s", fp) log.Printf("couldn't write to %s", fp)