adc23d5f96
git-subtree-dir: x git-subtree-mainline: 7d05a6ee8f44b314fa697a427439e5fa4d78c3d7 git-subtree-split: a10a11b9d371f36b7c3510da32a1d70b74e27bd1
29 lines
564 B
Go
29 lines
564 B
Go
package build
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
)
|
|
|
|
var (
|
|
defaultDir = sync.OnceValues(func() (string, error) {
|
|
dir := os.Getenv("OLLAMA_MODELS")
|
|
if dir == "" {
|
|
home, err := os.UserHomeDir()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
dir = filepath.Join(home, ".ollama", "models")
|
|
}
|
|
return dir, nil
|
|
})
|
|
)
|
|
|
|
// DefaultDir returns the default directory for models. It returns the value
|
|
// of the OLLAMA_MODELS environment variable if set; otherwise it returns
|
|
// "$HOME/.ollama/models".
|
|
func DefaultDir() (string, error) {
|
|
return defaultDir()
|
|
}
|